Вот мой код:
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]])
как я могу получить это?