python - Numpy 就地设置绝对值

Question

您有一个数组 A，您希望将其中的每个值都转换为绝对值。问题是

numpy.abs(A)

创建一个新矩阵，A 中的值保持原样。我找到两种方法将绝对值设置回 A

A *= numpy.sign(A)

或

A[:] = numpy.abs(A)

基于timeit测试，他们的表现相差无几

问题:

Are there more efficient ways to perform this task?

Answer 1

有一个 out 参数，它就地更新数组:

numpy.abs(A, out=A)

而且恰好是 很多 更快，因为您不必为新数组分配内存。

A = np.random.randn(1000, 1000)

%timeit np.abs(A)
100 loops, best of 3: 2.9 ms per loop

%timeit np.abs(A, out=A)
1000 loops, best of 3: 647 µs per loop