了解下 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)

大家都在看

发表评论

登录后才能评论
免费咨询
免费咨询
扫码关注
扫码关注
联系站长

站长Johngo!

大数据和算法重度研究者!

持续产出大数据、算法、LeetCode干货,以及业界好资源!

2022012703491714

微信来撩,免费咨询:xiaozhu_tec

分享本页
返回顶部