示例代码:
import numpy as np
a = np.array([[1,2,3],
[4,5,6]])
print(a[:,2])
#输出[3,6]
print(a[:,0:2])
#输出[[1 2]
[4 5]]
1、a[:,2]的意思就是取a的所有行的第2列元素
2、同理,a[1,:]的结果为[1,2,3]
3、a[:, m:n]即取矩阵a的所有行中的的第m到n-1列数据,含左不含右,就是包括m不包括n,如上面的a[:,0:2]就是取了a的第0列和第1列