如何從數組中創建1元素數組的數組？

Question

我想能夠轉換數組，如

a = np.array([[1,2], [3,4]]) 

到同一陣列，但各元件作爲1個元素的數組，而不是一個號碼。 所需的輸出將是：

np.array([[np.array([1]), np.array([2])], [np.array([3]), np.array([4])]]) 

Answer 1

您所描述的操作非常很少有用。更有可能的是，這將是一個更好的主意，長度爲1的一個額外的維度添加到您的數組的末尾：

a = a[..., np.newaxis] 
# or 
a = a.reshape(a.shape + (1,)) 

然後a[0, 1]將是一維數組，但所有的美好NumPy的功能，如廣播和ufuncs會工作的權利。請注意，這會創建原始數組的視圖;如果您需要獨立副本，則可以撥打copy()方法。

如果你真的想要一個二維數組，其元素是一維數組，NumPy不會讓你變得那麼容易。 （組織數據幾乎不是一種好的方法，所以NumPy開發人員沒有太多理由提供一種簡單的方法來完成它。）您可能希望創建這樣一個數組的大部分內容將創建一個3D陣列。最直接的辦法做到這一點我知道的是創建對象D型的空數組和填充細胞一個接一個，使用普通的Python循環：

b = numpy.empty(a.shape, dtype=object) 
for i in range(a.shape[0]): 
    for j in range(a.shape[1]): 
     b[i, j] = numpy.array([a[i, j]]) 

Answer 2

np.array([ [ [x] for x in row ] for row in [[1,2], [3,4]] ]) 

Answer 3

你的戰鬥np.array企圖創造一個高維陣列。

例如，你想要表達的，輸入的時候，產生了一個（2,2,1）陣列：

a2 = np.array([[np.array([1]), np.array([2])], [np.array([3]), np.array([4])]]) 

In [172]: a2 
Out[172]: 
array([[[1], 
     [2]], 

     [[3], 
     [4]]]) 

這實際上可能是你想要的（儘管印刷出場），因爲

In [180]: a2[0,0] 
Out[180]: array([1]) 

經常打的numpy的傾向提出更高的第二陣列的最好辦法，就是初始化它用正確的shape和dtype爲空，然後往裏面：

In [183]: a1=np.empty(a.shape,dtype=object)  
In [184]: a1.flat=[np.array(x) for x in a.flatten()] 
In [185]: a1 
Out[185]: 
array([[array(1), array(2)], 
     [array(3), array(4)]], dtype=object) 

它具有正確的形狀（2,2）和dtype。元素是數組 - 但它們的尺寸是「錯誤的」，而不是(1,)，而不是()。 flat和flatten是用於迭代nd數組的好用工具，就像它們是1d一樣。

a1.flat=[np.array([x]) for x in a.flatten()]不起作用 - 它是ints的一個數組，如a，但帶有對象dtype。

更明確的反覆做工作：

In [214]: a1=np.empty(a.shape,dtype=object) 
In [215]: for i in np.ndindex(2,2): 
    a1[i]=np.array([a[i]]) 
In [216]: a1 
Out[216]: 
array([[array([1]), array([2])], 
     [array([3]), array([4])]], dtype=object)