Как выбрать строку или столбец из матрицы? - PullRequest
0 голосов
/ 10 июня 2018

Здесь у меня есть матрица 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 ]

0 голосов
/ 10 июня 2018

Я думаю, нужно индексировать 0:3:

print (a[:,0:3])
[[ 1  2  3]
 [ 6  7  8]
 [11 12 13]]
0 голосов
/ 10 июня 2018

Попробуйте следующее

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 голосов
/ 10 июня 2018

Звучит так, как будто вы ищете a[:, 0:3]:

In [4]: a[:, 0:3]
Out[4]:
array([[ 1,  2,  3],
       [ 6,  7,  8],
       [11, 12, 13]])
...