从矩阵中得到某一列

设a是ndarray的矩阵,如果我要获得该矩阵的第k列并且以列向量的形式得到结果,那么我的第二维索引不能是一个数值,而应是一个范围。

例如如下代码获取3*3矩阵中的第二列

1 import numpy as np
2 import pandas as pd
3 a = np.array([[1,2,3],
4               [4,5,6],
5               [7,8,9]])
6 b = a[:, 1]
7 c = a[:, 1:2]
8 print(b)
9 print(c)

运行结果如下

猜你喜欢

转载自www.cnblogs.com/loubin/p/11296598.html