numpy中的一些统计函数往往需要指定轴axis
如:
np.sum(a,axis = 1)
那么这个axis=1实际上是说,只对第一轴进行相加操作。比如说:
a = np.array([[1,2,3],[1,2,3]])
print(np.sum(a,axis = 1))
>>>[6 6]
print(np.sum(a,axis = 0))
>>>[2 4 6]
print(np.sum(a))
>>>12
可以看到,当axis=0时,即相加只对第一维度,第一维度的数为[1,2,3]和[1,2,3],所以这两个矩阵相加为[2,4,6]。
当axis=1时,即相加只对第二维度,第二维度的数相加,第一维不变,即为[1+2+3,1+2+3]=[6,6]。
当不加axis参数时,把所有元素相加,即为12。