Как выбрать строку или столбец из матрицы?
Здесь у меня есть матрица a=np.array([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15]])
Я хочу выбрать все строки, но столбец, который я хочу выбрать, - от первого до третьего.
Так должно быть [[1,2,3],[6,7,8],[11,12,13]]
Тем не менее, я когда-либо пытался a[:,[0,2]]
, но это показывает
array([[ 1, 3],
[ 6, 8],
[11, 13]])
Это кажется неправильным, поэтому я попробовал еще один a[:][0:2], это все еще неверный результат.
Итак, я хочу спросить, есть ли какая-либо функция или метод может решить проблему?
3 ответа
Решение
Похоже, вы ищете a[:, 0:3]
:
In [4]: a[:, 0:3]
Out[4]:
array([[ 1, 2, 3],
[ 6, 7, 8],
[11, 12, 13]])
Попробуйте следующее
a=np.array([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15]])
a = a[:,0:3]
print(a)
#Output
#array([[ 1, 2, 3],
# [ 6, 7, 8],
# [11, 12, 13]])
Я думаю нужна индексация 0:3
:
print (a[:,0:3])
[[ 1 2 3]
[ 6 7 8]
[11 12 13]]