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]