多次元 NumPy 配列があります。
In [1]: m = np.arange(1,26).reshape((5,5))
In [2]: m
Out[2]:
array([[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10],
[11, 12, 13, 14, 15],
[16, 17, 18, 19, 20],
[21, 22, 23, 24, 25]])
と別の配列p = np.asarray([[1,1],[3,3]])
。p
のインデックスの配列として機能したかったm
、つまり:
m[p]
array([7, 19])
しかし、私は得る:
In [4]: m[p]
Out[4]:
array([[[ 6, 7, 8, 9, 10],
[ 6, 7, 8, 9, 10]],
[[16, 17, 18, 19, 20],
[16, 17, 18, 19, 20]]])
m
を使用して目的のスライスを取得するにはどうすればよいp
ですか?