分配给NumPy中的列？

Question

如何使用NumPy编写以下MATLAB代码？

A = zeros(5, 100);
x = ones(5,1);
A(:,1) = x;

分配行似乎很容易，但是我找不到将数组分配给另一个数组的列的示例。

Answer 1

使用a[:,1] = x[:,0]。您需要x[:,0]选择x作为单个numpy数组的列。如果可以选择格式化的方式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.]])