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

Это связано с тем, как срез работает в numpy, так как x[:,0] в вашем случае ожидает, что массив не является вектором.

Правильный способ достижения желаемого:

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

Answer 2

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