將一維數組放入二維數組

Question

我有一個2D數組x ，我想在其中復制1D數組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：無法將輸入數組從形狀（6）廣播到形狀（3,2）

這個怎么做 ？

Answer 1

您必須將y轉換為形狀類似於x的數組：

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

但是請注意， y應該用x的形狀重塑。