陣列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)

但是對於像這樣的情況，我會使用其他方法之一。