将计算结果存储在numpy数组中（而不是创建新的数组）

Question

假设我有以下代码：

a, b, c = np.empty((3,)), np.array([1, 2, 3]), np.array([4, 5, 6])
a = b + c

我想要将b + c的结果存储在现有数组a中。 我不希望分配新数组。 我怎样才能在numpy中做到这一点？

Answer 1

np.add(b, c, out=a)

NumPy ufuncs（和其他一些NumPy例程）采用out参数将输出放入。

Answer 2

如果您的内存至关重要，并且您不想漫游3个数组（a，b和c），则可以执行以下操作：

b, c = np.array([1, 2, 3]), np.array([4, 5, 6])
b += c  # does in-place operation, similarly to np.add(b, c, out=b)

其中b现在携带您在数组a获得的结果。