1. 程式人生 > >Numpy中axis的理解

Numpy中axis的理解

numpy中的一些統計函式往往需要指定軸axis
如:
np.sum(a,axis = 1)
那麼這個axis=1實際上是說,只對第一軸進行相加操作。比如說:

a = np.array([[1,2,3],[4,5,6]])
print(np.sum(a,axis = 1))
>>>[6 15]
print(np.sum(a,axis = 0))
>>>[5 7 9]
print(np.sum(a))
>>>21

可以看到,當axis=0時,即相加只對第一維度,第一維度的數為[1,2,3]和[4,5,6],所以這兩個矩陣相加為[5,7,9]。
當axis=1時,即相加只對第二維度,第二維度的數相加,第一維不變,即為[1+2+3,4+5+6]=[6,15]。
當不加axis引數時,把所有元素相加,即為21。