Python 在同一行中获取 Numpy 相等的结果

Question

我确定有这个功能，但我找不到。

我正在做如下计算:

# L = large iterator of lots of numpy arrays
def replace_error_nan(data):
    data[data == ERROR_VALUE] = np.nan
    return data
L = (replace_error_nan(l) for l in L)

我想执行 replace_error_nan 完成的所有操作而不需要使用该函数。我敢肯定这是非常简单的事情。

谢谢!

---

问题已被留下“小”评论的人回答。我给自己做了一个回答。谢谢!

Answer 1

没有错:

for l in L:
  l[l == ERROR_VALUE] = np.nan

如果你坚持单行，你可以在理解中使用__setitem__。但是不要。