У меня есть массив как:
a = array([[1,2,3],[3,4,5],[4,5,6]])
что самый эффективный путь состоит в том, чтобы нарезать 1x2 массив из этого, которое имеет только первые два столбца "a"?
Т.е.
array([[2,3],[4,5],[5,6]]) in this case.
спасибо.
Двумерные массивы numpy индексируются с использованием a [i, j]
(не a [i] [j]
), но вы можете использовать ту же нотацию срезов с множеством массивов и матриц, что и с обычными матрицами в Python (просто поместите их в один []
):
>>> from numpy import array >>> a = array([[1,2,3],[3,4,5],[4,5,6]]) >>> a[:,1:] array([[2, 3], [4, 5], [5, 6]])