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] 

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