Numpy中关于axis的理解
程序员文章站
2024-03-02 16:04:40
...
一个维度为n的数组,它的axis的取址范围为[0,n-1],numpy中有sum()
,min()
,max()
,mean()
,std()
,var
…等函数,我们要根据不同的需求使用不同的axis。
x = np.arange(24).reshape(2,3,4)
print(x.sum(axis=0))
[[12 14 16 18]
[20 22 24 26]
[28 30 32 34]]
print(x.sum(axis=1))
[[12 15 18 21]
[48 51 54 57]]
print(x.sum(axis=2))
[[ 6 22 38]
[54 70 86]]
如上述结果所示,axis的取址决定将数组中的那些元素放在一起统计,所以我们可以得到这样的结论:指定统计的axis的下标改变,其他轴下标不变,将这样的数据放在一起统计。