axis=0, 1该如何形象的理解呢？

Question

bobo老师，我经常会把axis=0/1搞错。你是怎么防止自己出现这样的问题的呢？

谢谢。

Answer 1

liuyubobobo 2018-01-14 01:55:00

我记得在这个课程中第一次遇到axis容易混淆的时候我提过：

大多数axis=x，这里的x是指压缩掉的维度，也就是仍掉的维度。比如np.sum(X, axis=1)，是压缩掉第1维度，只剩下第0维度，也就是对每一行进行求和，我习惯这么记忆；

另一种方式就是大多数axis=x表示“沿着”这个维度做事情（在文档中用along），比如np.sum(X, axis=1)，是沿着第1维度的方向求和，也就是沿着列方向求和，也就是对每一行求和。

• 提问者 爱赵晓羊 #1

  非常感谢bobo老师