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

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

新規登録して質問してみよう
ただいま回答率
85.50%
NumPy

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

Python 3.x

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

Python

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

Q&A

解決済

2回答

2507閲覧

Python:Numpy配列の変形

piyopiyp

総合スコア8

NumPy

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

Python 3.x

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

Python

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

0グッド

0クリップ

投稿2017/01/03 08:54

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

python

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

のように書けます.

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

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

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

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

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

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答2

0

ベストアンサー

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

python

1import numpy 2 3x = numpy.arange(15).reshape(5, 3) 4indices = numpy.array([0, 1, 2, 1, 2]) 5y = x[numpy.arange(5), indices] 6 7# print -> [ 0 4 8 10 14] 8print(y)

投稿2017/01/03 10:50

nukui-s

総合スコア13

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

piyopiyp

2017/01/04 04:56

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

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/03 10:03

sartan

総合スコア14

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

ozwk

2017/01/04 05:07

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

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

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

ただいまの回答率
85.50%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問