Numpy, заменить столбец с массивом

Question

Я пытаюсь заменить один или несколько столбцов новым массивом такой же длины.

a = np.array([[1,2,3],[1,2,3],[1,2,3]])
b = np.array([[0,0,0])
a[:, 0] = b

Я получил ошибку ValueError: could not broadcast input array from shape (3,1) into shape (3).Однако это работает, когда b имеет несколько столбцов.

a = np.array([[1,2,3],[1,2,3],[1,2,3]])
b = np.array([[0,7],[0,7],[0,7]])
a[:, 0:2] = b

array([[0, 7, 3],
       [0, 7, 3],
       [0, 7, 3]])

Как эффективно заменить столбец другим массивом?

Спасибо

J

Answer 1

Ваш пример будет работать нормально, если вы используете следующее так же, как вы используете a[:, 0:2] = b. [:, 0:1] фактически является только первым столбцом

a = np.array([[1,2,3],[1,2,3],[1,2,3]])
b = np.array([[0],[0],[0]])
a[:, 0:1] = b

# array([[0, 2, 3],
#        [0, 2, 3],
#        [0, 2, 3]])

Answer 2

У вас неправильная форма b.Вы должны передать ему обычный одномерный массив, если хотите заменить только один столбец:

a = np.array([[1,2,3],[1,2,3],[1,2,3]])
b = np.array([0,0,0])
a[:, 0] = b
a

Возвращает:

array([[0, 2, 3],
       [0, 2, 3],
       [0, 2, 3]])