Numpy：多次元配列の複数の要素の取得

前提・実現したいこと

多次元配列から、複数の要素を取得したい。
二次元配列の例：

Python
1import numpy as np
2
3a = np.arange(12).reshape(3, 4)
4b = np.array([[1, 0], [2, 1], [3, 2]])
5
6#aから、bの複数の座標の要素を取得
7
8result = [1, 6, 11] #求めている結果

発生している問題・エラーメッセージ

上記の「#aから、bの複数の座標の要素を取得」の操作方法
下記の試したことでもできなくはないが、他の多次元配列への汎用性がなく、
二次元配列上での要素範囲を超えても、エラーにならない場合がある。
正攻法があるのであればご教授願いたい。

試したこと

一次元配列の場合は、a[b]の操作でできるので、

Python
1import numpy as np
2
3#一次元配列の場合
4a = np.arange(12)
5b = np.array([5,1,2,4])
6a[b] #>>>[5,1,2,4]
7
8#一次元配列で可能なことを活用した二次元配列の複数の要素の取得
9c = np.arange(12).reshape(3, 4)
10d = np.array([[1, 0], [2, 1], [3, 2]])
11c.flatten()[d@[1, c.shape[1]]] #>>>[1, 6, 11]

補足情報（FW/ツールのバージョンなど）

Python 3.9.2
Numpy 1.20.2
Windows 10

行動規範の内容に同意します

回答3件

ベストアンサー

python
1a[tuple(b.T[::-1])]

とか。
(0次元目と1次元目が入れ替わっているのはわざとなんでしょうか？)

投稿2021/05/21 15:05

bsdfan

総合スコア4794

sakuramochi_py

2021/05/22 07:33

すみません！０、１次元が入れ替わっているのは私のミスです！０次元目と1次元目に分けることで取得しているんですね！差し支えなければ、tupleに変換する理由なども教えていただけないでしょうか。

bsdfan

2021/05/22 08:47 編集

a[x, y, z]はa[(x, y, z)]と同じです。これはnumpyではなく、pythonの[](__getitem__())の仕様です。 https://qiita.com/gyu-don/items/bde192b129a7b1b8c532 tuple(b)は(b[0], b[1])で、 tuple(b.T)は(b[:, 0], b[:, 1])になります。なのでa[tuple(b.T)]はa[b[:, 0], b[:, 1]]と同じになります。今回は[::-1]を追加して次元の入れ替えをしています。結果として、hayataka2049さんのと同じになります。

sakuramochi_py

2021/05/22 09:03

なるほど！tupleはpythonの仕様を活用したものだったんですね！ご丁寧にありがとうございます！

行動規範の内容に同意します

正攻法かどうかは分かりませんが、以下の式で何次元でもできます。

python
1>>> a = np.arange(12).reshape(3, 4)
2>>> b = np.array([[1, 0], [2, 1], [3, 2]])
3>>> result = [a.__getitem__(tuple(c[::-1])) for c in b]
4>>> print(result)
5[1, 6, 11]