5

我偶尔会where在 numpy 的 ufunc 中使用该子句。例如,以下内容:

import numpy as np
a = np.linspace(-1, 1, 10)
np.sqrt(a, where=a>0) * (a>0)

在 Numpy 1.12 及更早版本中,这曾经在可能的情况下给我平方根值,否则为零。

不过,最近我升级到了 numpy 1.13。上面的代码现在给了我以下错误:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: Automatic allocation was requested for an iterator operand, and it was flagged as readable, but buffering  without delayed allocation was enabled

我认为这正是该where子句应该使用的方式,但也许我错了。所以我有两个问题:第一,这段代码有什么问题;其次,实现我的目标的推荐方法是什么?

4

1 回答 1

3

供将来参考:这原来是 numpy 中的一个错误。它在下一个 numpy 版本中修复,大概是 1.13.1 版。

1.13.0 的解决方法是显式为 ufunc 提供out参数。在上面的示例中,np.sqrt(a, where=a>0, out=np.zeros(a.shape))有效。

于 2017-07-01T13:40:06.190 回答