阵列python的形状

Question

假设我创建了一个二维数组

 m = np.random.normal(0, 1, size=(1000, 2))
 q = np.zeros(shape=(1000,1))
 print m[:,0] -q

当我接受时,m[:,0].shape我得到的(1000,)不是(1000,1)我想要的.我如何强制m[:,0]进入(1000,1)阵列？

Answer 1

通过选择第0列,正如您所注意到的那样,您可以减少维度:

>>> m = np.random.normal(0, 1, size=(5, 2))
>>> m[:,0].shape
(5,)

你有很多选择可以让5x1对象退出.您可以使用列表而不是整数进行索引:

>>> m[:, [0]].shape
(5, 1)

您可以要求"所有列,但不包括1":

>>> m[:,:1].shape
(5, 1)

或者您可以使用None(或np.newaxis),这是扩展维度的一般技巧:

>>> m[:,0,None].shape
(5, 1)
>>> m[:,0][:,None].shape
(5, 1)
>>> m[:,0, None, None].shape
(5, 1, 1)

最后,你可以重塑:

>>> m[:,0].reshape(5,1).shape
(5, 1)

但是对于像这样的情况,我会使用其他方法之一.