Назначение столбцам в NumPy?
Как следующий код MATLAB может быть написан с использованием NumPy?
A = zeros(5, 100);
x = ones(5,1);
A(:,1) = x;
Назначение строк, кажется, работает легко, но я не смог найти пример назначения массива столбцу другого массива.
Ответы на вопрос(2)
Решение Вопроса
Используйтеa[:,1] = x[:,0]. Тебе нужноx[:,0], чтобы выбрать столбец х в виде единого массива. Если у вас есть выбор, как отформатировать x, лучше сначала не делать его двумерным массивом, а просто обычным (строчным) массивом:
>>> a
array([[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]])
>>> x = numpy.ones(5)
>>> x
array([ 1., 1., 1., 1., 1.])
>>> a[:,1] = x
>>> a
array([[ 0., 1., 0.],
[ 0., 1., 0.],
[ 0., 1., 0.],
[ 0., 1., 0.],
[ 0., 1., 0.]])
a[:,1] означает все элементы столбца 1. См. numpy Базовая нарезка и индексирование.