將一維數組爲二維數組

Question

我有一個二維數組x中，我想將內容複製一維數組y的：

import numpy as np 
x = np.array([[1, 2], [4, 5], [3, 3]], np.int32) 
y = np.array([1, 2, 3, 4, 5, 6]) 

x[:,:] = y # i would like x to be [[1, 2], [3, 4], [5, 6]] 

ValueError: could not broadcast input array from shape (6) into shape (3,2)

如何做到這一點？

Answer 1

你必須在y轉換爲陣列的形狀像x：

>>> x = y.reshape(x.shape) 
>>> x 
array([[1, 2], 
     [3, 4], 
     [5, 6]]) 

但要注意的是，y應x的形狀重塑。