質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

90.83%

  • Python

    5950questions

    Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

  • Python 3.x

    4477questions

    Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

  • NumPy

    326questions

    NumPyはPythonのプログラミング言語の科学的と数学的なコンピューティングに関する拡張モジュールです。

Python:Numpy配列の変形

解決済

回答 2

投稿

  • 評価
  • クリップ 0
  • VIEW 615

piyopiyp

score 2

PythonでNumpy配列を変形する方法についてお聞きしたいです.
Numpyで配列にmax関数を適用すると,指定した軸ごとの最大値が得られます.
例)
イメージ説明
これは,行ごとの最大値をとって新しい配列にする例で,

np.max(x, axis=1)#xは入力配列


のように書けます.

しかし,私がしたい変形は,最大値ではなく,指定したインデックスの値を抽出して新しい配列にする変形です.
イメージは以下の通りです.
イメージ説明
このように,入力配列と,インデックスの入った配列が与えられたとして,そこからmax関数のように新しい配列を作るためには,どのようにコードを書けばいいでしょうか.

極力for文を使わない実装を目指しています.
ご教授お願いいたします.

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 2

checkベストアンサー

0

numpyのindexingを利用すれば以下のように簡潔に書けます:

import numpy

x = numpy.arange(15).reshape(5, 3)
indices = numpy.array([0, 1, 2, 1, 2])
y = x[numpy.arange(5), indices]

# print -> [ 0  4  8 10 14]
print(y)

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2017/01/04 13:56

    出来ました!ありがとうございます。

    キャンセル

0

numpyには最大値のindexを返してくれるargmax関数というものがあります。
それを使えば良いと思います。
for文は一度使っていますが、以下のようなコードになりました。

import numpy as np

x=np.random.randint(0,7,32).reshape(8,4)
n = np.array([np.argmax(i) for i in x])
print(n)

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2017/01/04 14:07

    私も最初読み間違えたんですが
    行列の行ごとにインデックスを指定し、それを抽出したいのであって
    最大値のインデックスを求めたいわけではないようです。

    キャンセル

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 90.83%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

  • 解決済

    xcodeでの画面サイズ変更方法

    swiftでサンプルコードを写しているのですが、参考書を今まで放置してたのでxcodeのバージョンが違うので画面サイズ変更の方法がわからないです。 simulatorでの表示画面

  • 解決済

    Python:PILで画像を保存したい

    Pythonの画像処理ライブラリ,Pillowを使って,配列から画像を作成し,それを保存したいです. try: import urllib.request excep

  • 解決済

    Python2.7.10のIndexError: only integers, sliceの対処方

    前提・実現したいこと Python2.7.10を使用してparticlefilterを実装したいです. python + numpyで緑色の物体を追跡する (パーティクルフ

  • 解決済

    PyQtにおけるQWidgetとQGraphicsViewの組み合わせ

    http://d.hatena.ne.jp/mFumi/20141112/1415806010 このサイトをお手本にQWidgetとQGraphicsViewの組み合わせを実

  • 解決済

    回答ありがとうございました

    質問消せないようなので問題文は 消させていただきます 回答ありがとうございました

  • 解決済

    vim-flake8を導入した際エラー行の色を変えたい

    実現したいこと macのデフォルトターミナル(novel)上でvimを開き、 nvie/vim-flake8のプラグインを導入しました。 F7を押すとpep8違反となる行を表

  • 解決済

    pythonの図示 matpotlib

    pythonで四角形が格子状に区切られそれぞれの点に数字が振られています。それをコンターマップのように(ただし、等高線はいらない)図示したいのですがどのように書けばよいでしょうか

  • 受付中

    matplotlibでの軸ラベルのフォント

    matplotlibを用いてグラフの描画をしたいと思っています。 そこで、軸ラベルのフォントをアルファベットの場合にはArial, ギリシャ文字の場合にはSymbolと使い分けたい

同じタグがついた質問を見る

  • Python

    5950questions

    Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

  • Python 3.x

    4477questions

    Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

  • NumPy

    326questions

    NumPyはPythonのプログラミング言語の科学的と数学的なコンピューティングに関する拡張モジュールです。