numpy Python冒号和省略号索引的区别

cwtwac6a  于 2023-01-20  发布在  Python
关注(0)|答案(1)|浏览(145)

我一直在试验使用冒号和省略号的Numpy数组索引。但是,我不能理解我得到的结果。
下面是示例代码:

>>> a = np.array([[1,2],[3,4]])
>>> a
array([[1, 2],
       [3, 4]])

>>> a[:,np.newaxis]     #  <-- the shape of the rows are unchanged
array([[[1, 2]],

       [[3, 4]]])
>>> a[...,np.newaxis]   #  <-- the shape of the rows changed from horizontal to vertical
array([[[1],
        [2]],

       [[3],
        [4]]])
9udxz4iz

9udxz4iz1#

原是(2,2)
加上:,它就变成了(2,1,2)。新的轴添加在第一个维度之后。
使用...,形状为(2,2,1),新形状最后添加。

相关问题