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]