如何理解数组的行数和列数，差点搞蒙？

最近接触数组，就这个行和列有时候傻傻分不清楚，因为维度太高。今天看一个简单的小例子，即可搞懂咋回事！

import numpy as np
np.random.seed(10)
a=np.random.randint(1,10,[5,3])
print(a,np.amax(a,axis=0),a.argmax(axis=0))  
print(a,np.amax(a,axis=1),a.argmax(axis=1))

输出：

a=[[5 1 2]
 [1 2 9]
 [1 9 7]
 [5 4 1]
 [5 7 9]]

a是一个5行3列的二维数组。

np.amax(a,axis=0)=[5 9 9]   #axis=0,表示取出每列的最大值
a.argmax(axis=0))=[0 2 1]   #每列最大值的索引
np.amax(a,axis=1)=[5 9 9 5 9]  #axis=1，表示取出每行的最大值
a.argmax(axis=1)= [0 2 1 0 2]   #每行最大值的索引

print(a.sum(axis=1))  #[ 8 12 17 10 21]  axis=1表示对每行求和
print(a.sum(axis=0))  #[17 23 28]        axis=0表示对每列求和

print(np.size(a,0))   # 5 返回其行数
print(np.size(a,1))   # 3 返回其列数

看了这个例子，应该清楚了吧。

import numpy as np
np.random.seed(10)
a=np.random.randint(1,10,[5,3])
print(a,np.amax(a,axis=0),a.argmax(axis=0))  
print(a,np.amax(a,axis=1),a.argmax(axis=1))
print(a.sum(axis=1)) 
print(a.sum(axis=0))  

至于输出结果自己运行下，就理解了。