了解下 numpy 三维布局

a = np.arange(24).reshape(3,2,4)

看上面这个numpy形成的三维数组!

深度:3

高度:2

宽度:4

下面画了一个三维图示,大家用来理解!

了解下 numpy 三维布局

先不说高纬度,就看三维,有一些同学这几天问到我,看了上图应该会有一个清晰的理解。

然后下面再理解下 axis 这个参数的功能:

>>> np.sum(a, axis=0)
array([[24, 27, 30, 33],
       [36, 39, 42, 45]])
>>> np.sum(a, axis=1)
array([[ 4,  6,  8, 10],
       [20, 22, 24, 26],
       [36, 38, 40, 42]])
>>> np.sum(a, axis=2)
array([[ 6, 22],
       [38, 54],
       [70, 86]])

axis = 0 代表了深度方向

axis = 1 代表了高度方向

axis = 2 代表了宽度方向

也对应了 reshape(3, 2, 4) 参数的顺序!

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/5184/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

最近整理资源【免费获取】:   👉 程序员最新必读书单  | 👏 互联网各方向面试题下载 | ✌️计算机核心资源汇总