Numpy 3d array удаляет первую запись из каждой строки - PullRequest
0 голосов
/ 02 марта 2019

Допустим, у меня есть массив 3d Numpy:

array([[[0, 1, 2],
        [0, 1, 2],
        [0, 2, 5]]])

Можно ли удалить первую запись из всех строк (этих самых внутренних строк).В этом случае 0 будет удалено в каждой строке.

Дает нам следующий вывод:

[[[1, 2],
  [1, 2],
  [2, 5]]]

Ответы [ 2 ]

0 голосов
/ 02 марта 2019

В дополнение к ответу @ coldspeed), нарезка в numpy является очень мощной и может быть выполнена различными способами, включая оператор двоеточия : в индексе , то есть

print(x[:,:,1:])
# array([[[1, 2],
#         [1, 2],
#         [2, 5]]])

эквивалентно установленному использованию многоточия.

0 голосов
/ 02 марта 2019
x
array([[[0, 1, 2],
        [0, 1, 2],
        [0, 2, 5]]])

x.shape
# (1, 3, 3)

Вы можете использовать Ellipsis (...) для выбора по всем крайним осям и вырезать первое значение из каждой строки с помощью 1:.

* 1008.*
...