我一直在试验使用冒号和省略号的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]]])
1条答案
按热度按时间9udxz4iz1#
原是(2,2)
加上:,它就变成了(2,1,2)。新的轴添加在第一个维度之后。
使用...,形状为(2,2,1),新形状最后添加。