Python numpy slice with None: expand dims

最新推荐文章于 2022-03-04 08:59:12 发布

HeroFUCKEVERYTHING

最新推荐文章于 2022-03-04 08:59:12 发布

阅读量187

点赞数

分类专栏： python-numpy

本文链接：https://blog.csdn.net/Hero_Never_GIVE_UP/article/details/115247651

版权

python-numpy 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

Python numpy 使用“None”进行切片时可以扩增数据元素的维度，效果等同于于np.newaxis()

>>> b = np.arange(10)
>>> b
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> 
>>> 
>>> b[None]
array([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]])
>>> b[:,None]
array([[0],
       [1],
       [2],
       [3],
       [4],
       [5],
       [6],
       [7],
       [8],
       [9]])

None 的位置决定了在哪个维度上扩增。第一个位置扩增1轴，第二个位置扩增2轴