индексирование многомерных массивов с помощью массива
У меня есть многомерный массив 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
?