3

多次元 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ですか?

4

1 に答える 1