a = np.arange(20).reshape(4,5)
[[ 0 1 2 3 4]
[ 5 6 7 8 9]
[10 11 12 13 14]
[15 16 17 18 19]]
获取二维数组元素
print a[1, 3] #8
print a[1:3, 3:5] #[[ 8 9][13 14]]
print a[1, :] #[5 6 7 8 9]
行或列进行向量化运算
print a[0, :] + a[1, :] #[ 5 7 9 11 13]
print a[:, 0] + a[:, 1] #[ 1 11 21 31]
获取第一行最大数
print a[0,:].max() #4
获取第一行最大数的索引
print a[0,:].argmax() #4
np.argmax(a, axis=0) #0代表列 array([3, 3, 3, 3, 3], dtype=int64)
np.argmax(a, axis=1)#1代表行 array([4, 4, 4, 4], dtype=int64)
获取5列的平均值
print a[:,4].mean() #11.5
axis=0代表列,axis=1代表行。下面获取列的平均值和行的sum值:
print a.mean(axis=0) #[ 7.5 8.5 9.5 10.5 11.5]
print a.sum(axis=1) #[10 35 60 85]