Вот пример:
t = np.random.randint(10, 20, size=(2, 3, 5))
array([[[11, 16, 13, 18, 19],
[13, 18, 14, 19, 15],
[12, 16, 12, 15, 12]],
[[18, 16, 12, 11, 19],
[14, 18, 13, 16, 17],
[18, 17, 16, 16, 10]]])
Я хотел бы выбрать массивы [0, 1,:] и [1, 2,:],
, которые имеют следующую форму : (2, 5)
[[13, 18, 14, 19, 15],
[18, 17, 16, 16, 10]]
Я попробовал приведенный ниже код, но он не даст мне правильного ответа
t[:, [1, 2], :]
array([[[13, 18, 14, 19, 15],
[12, 16, 12, 15, 12]],
[[14, 18, 13, 16, 17],
[18, 17, 16, 16, 10]]])
Я могу выполнить итерацию, чтобы получить массивы и соединить их с сделать форму, которую я хотел, но мне интересно, есть ли способ получить правильные массивы, используя индексы.
Спасибо.