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)

不过在这种情况下，我会选择其他方法。