Python：將向量插入矩陣

Question

我有以下代碼：

import numpy as np
x = np.zeros([4,N])
x[:,0]= np.vstack([1000,0,0,50])

但是，出現以下錯誤：

ValueError: could not broadcast input array from shape (4,1) into shape (4)

我很困惑為什么它不起作用，任何幫助將不勝感激。

Answer 1

x[:,0]= np.array([1000,0,0,50]).T

Answer 2

這是由於在numpy中切片的工作方式為x[:,0]在您的情況下期望數組不是向量。

實現所需目標的正確方法是：

import np as numpy
x = np.zeros([4,N])
x[:,0]= np.array([1000,0,0,50])