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])