Numpy как получить выборочные массивы в массивах - PullRequest
0 голосов
/ 26 марта 2020

Вот пример:

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]]])

Я могу выполнить итерацию, чтобы получить массивы и соединить их с сделать форму, которую я хотел, но мне интересно, есть ли способ получить правильные массивы, используя индексы.

Спасибо.

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