Numpy按轴计算

a = np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])

如果二维数组a展示到图上就是:

因为a[0]为[1,2,3,4],a[1]为[5,6,7,8],a[2]为[9,10,11,12],所以[1,2,3,4]在0轴0的位置,[5,6,7,8]在0轴1的位置,而[9,10,11,12]在0轴2的位置。

1是a[0][0],2是a[0][1], 3是a[0][2],4是a[0][3],所以[1,2,3,4]按照自己的下标顺序沿着1轴依次展开。5,6,7,8和9,10,11,12也是同样道理,按照自己的下标顺序沿着1轴依次展开。

数组计算时,声明axis=0,就时沿着0轴方向进行;axis=1,就是沿着1轴方向进行。

例如:a.sum(axis=0)

计算就是沿着0轴方向进行元素求和,也就是计算每列的和:

最终得到的数组是[15,18,21,24]

 

再比如 a.mean(axis=1) 

计算就是沿着1轴方向进行元素求平均值,也就是计算每行的平均值:

最终得到的数组是[2.5, 6.5, 10.5] 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章