取消python-xarray的警告

Question

我正在运行以下代码

positive_values = values.where(values > 0)  

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

RuntimeWarning: invalid value encountered in greater_equal if not reflexive  

xarray是否有压缩这些警告的方法？

Answer 1

作为一个一般的经验法则，警告应该被注意而不是被压制。要么你知道是什么引起了警告，要么你不知道。如果您知道引起警告的原因，通常有一个简单的解决方法。如果您不知道是什么原因导致警告，则很可能存在bug。在这种情况下，您可以使用&的短路特性，如下所示:

positive_values = values.where(values.notnull() & values > 0)

Answer 2

warnings模块提供您要查找的功能。

要取消所有警告，请执行以下操作（请参阅John Coleman的答案，了解为什么这不是一个好的实践）:

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)  

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