1. 程式人生 > >numpy 軸的一些理解

numpy 軸的一些理解

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。