如何有效地设置1D数组的numpy数组2D整列？

Question

我有一个NxM维度的numpy数组“A”和N维的M个数组“B”。 在for（i：0,1，...... M）中，我正在计算这个B数组，我想设置A数组的整个列“i”。 我试图用A [：，i]访问，但效果太慢。

如何有效地做到这一点？

# Initialize array.
A = np.empty([N, M])

for i in range(M):
    # Compute of B (array N elements):
    B = ...An operation that results in a 1D array, different in each iteration
    A[:, i] = B

编辑：B值与“i”次迭代不同。