沿着某个维度进行操作

一直对沿着axis进行操作不太理解，今天决定还是要搞明白。

比如np.sum(a, aixs=1)，代表沿着1维操作。这个沿着指的是下标变化的方向，其他维度不动a[0][0],a[0][1], a[0][2]​。

于是将红色框的元素相加得到：

1
2
3

[6],
[9],
[16]

如果是np.sum(a, axis=0)，即a[0][0],a[1][0], a[2][0]

1

[6],[9],[16]

参考链接：

https://zhuanlan.zhihu.com/p/31275071