numpy & axis

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