Как выбрать строку или столбец из матрицы?

Здесь у меня есть матрица 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]]
Другие вопросы по тегам