Python中warnings.warn仅显示一次警告的问题及解决咨询

原因 #

Python的warnings模块默认带有警告过滤与去重机制：它会记录已经发出过的警告的类别和消息内容，当相同的警告再次被触发时，会自动跳过输出，避免重复打扰用户。在Jupyter Notebook环境中，多次执行同一代码块时，第一次发出的警告会被标记为已触发，后续即使代码执行到warnings.warn语句，模块也会过滤掉重复警告。

解决办法 #

以下几种方式可以实现每次执行都显示警告：

1. 每次调用前重置警告状态 #

使用warnings.resetwarnings()清除模块记录的所有已触发警告信息，让后续的warn调用重新输出警告：

import warnings

class MyClass:
    def my_method(self):
        print('Here')
        warnings.resetwarnings()
        warnings.warn("This is a warning message")
        print('Still here!')

2. 修改警告过滤器为强制输出 #

通过warnings.simplefilter("always")设置全局过滤器，让所有警告始终被输出：

import warnings

# 全局设置：所有警告都强制输出
warnings.simplefilter("always")

class MyClass:
    def my_method(self):
        print('Here')
        warnings.warn("This is a warning message")
        print('Still here!')

如果不想全局生效，可以用上下文管理器临时修改过滤器，避免影响其他警告逻辑：

import warnings
from contextlib import contextmanager

@contextmanager
def temporary_always_warn():
    # 保存原过滤器配置
    original_filters = warnings.filters.copy()
    warnings.simplefilter("always")
    yield
    # 恢复原配置
    warnings.filters = original_filters

class MyClass:
    def my_method(self):
        print('Here')
        with temporary_always_warn():
            warnings.warn("This is a warning message")
        print('Still here!')

3. 使用自定义警告类别（适合区分场景） #

如果需要针对不同场景的警告，自定义一个Warning子类，这样即使消息相同，不同类别的警告也不会被过滤：

import warnings

# 自定义警告类别
class MyCustomWarning(Warning):
    pass

class MyClass:
    def my_method(self):
        print('Here')
        warnings.warn("This is a warning message", MyCustomWarning)
        print('Still here!')

注意：如果同一自定义类别的相同消息被重复触发，仍然会被过滤，此方法更适合区分不同逻辑的警告场景。

内容的提问来源于stack exchange，提问作者Britzel