python - 抑制 python-xarray 的警告

Question

我正在运行以下代码

positive_values = values.where(values > 0)  

在此示例中，values 可能包含 nan 元素。我相信由于这个原因，我收到了以下运行时警告:

RuntimeWarning: invalid value encountered in greater_equal if not reflexive  

xarray 是否有抑制这些警告的方法？

Answer 1

warnings模块提供了您正在寻找的功能。

要抑制所有警告(参见 John Coleman's answer 为什么这不是好的做法):

import warnings
warnings.simplefilter("ignore") 
# warnings.simplefilter("ignore", category=RuntimeWarning) # for RuntimeWarning only

要使抑制暂时在 warnings.catch_warnings() 上下文管理器中执行:

import warnings

with warnings.catch_warnings():
    warnings.simplefilter("ignore")
    positive_values = values.where(values > 0)  

上下文管理器在进入上下文之前保存原始警告设置，然后在退出上下文时将它们恢复原状。