numpy newaxis并不总是在我想要的位置追加新的轴

Question

为什么np.newaxis总是追加我放置的轴？见例如：

import numpy as np 
M = np.array([[1,4],[2,5], [3,4]]) 
M[:,np.newaxis].shape 
# returns (3, 1, 2) 

Answer 1

按照documents描述newaxis：

添加的尺寸是在所述选择元组中的newaxis对象的位置。

在您的示例中，newaxis位于元组的第二个位置，因此插入了长度为1的新维度。

这类似于在特定索引处选择一个值。对于三维A，您可以使用A[:,0]从第二个轴检索索引0值，而不是第三个轴。

如果要添加新中轴线的元组的最后一个位置，你可以写M[:,:,np.newaxis]或交替使用省略号表示法：

>>> M[...,np.newaxis].shape 
(3, 2, 1)