答案：UserWarning用于提示非错误但需注意的情况，通过warnings.warn()抛出，可用filterwarnings()控制显示或转为异常，结合catch_warnings()可捕获用于测试。
在Python中，UserWarning 是一种用于提醒用户代码中潜在问题的警告类型，通常由开发者主动触发，提示使用者注意某些非错误但可能影响结果的情况。处理自定义的 UserWarning 包括两个方面：如何抛出它，以及如何控制它的显示或捕获它。
抛出自定义UserWarning
使用 warnings.warn() 函数可以手动触发一个 UserWarning。你可以传入警告信息和警告类别。
- import warnings
- def divide(a, b):
- if b == 0:
- warnings.warn("除数为0，返回0", UserWarning)
- return 0
- return a / b
调用 divide(10, 0) 时会输出警告信息，但程序不会中断。
控制UserWarning的显示
默认情况下，Python 会显示所有警告一次。你可以通过命令行或在代码中使用 warnings.filterwarnings() 来控制警告行为。
立即学习“Python免费学习笔记（深入）”；
- # 忽略所有UserWarning
- warnings.filterwarnings("ignore", category=UserWarning)
- # 只显示一次（默认）
- warnings.filterwarnings("once", category=UserWarning)
- # 每次都显示
- warnings.filterwarnings("always", category=UserWarning)
- # 将警告转为异常
- warnings.filterwarnings("error", category=UserWarning)
例如，将警告转为异常后，触发 UserWarning 会导致程序抛出异常并终止执行。
捕获和测试UserWarning
在编写测试代码时，你可能想验证某个函数是否正确发出了警告。可以使用 warnings.catch_warnings() 上下文管理器来捕获警告。
- import warnings
- with warnings.catch_warnings(record=True) as w:
- warnings.simplefilter("always")
- divide(5, 0)
- if len(w) > 0:
- print("捕获到警告:", w[0].message)
这个方法常用于单元测试中，确保警告逻辑按预期工作。
基本上就这些。合理使用 UserWarning 能提升代码的可维护性和用户体验，同时通过过滤机制灵活控制警告行为。不复杂但容易忽略细节。
