Как я могу получить элемент из массива 3d numpy через массив двумерных индексов? - PullRequest
0 голосов
/ 01 апреля 2020

Вот мой код:

import numpy as np


arr = np.arange(27).reshape(3, 3, 3)
indices = np.array([[0, 2], [1, 2], [2, 0]]) 

arr
# array([[[ 0,  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, 26]]]) 

indices
# array([[0, 2],
#       [1, 2],
#       [2, 0]])

Я хотел бы получить предметы из arr по индексу indices, например:

array([[ 6,  7,  8],
       [15, 16, 17],
       [18, 19, 20]])

как я могу получить это?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...