python捕获警告的三种方法

为了让读者更好地了解捕获警告的方式，下面将从以下三个方面进行讲解：

1. 捕获警告的基本概念
2. Python捕获警告的三种方法
3. 两个示例说明

一、捕获警告的基本概念

在 Python 中，警告是一种异常情况，可以被捕获和处理，常见的有以下几种情况：

• DeprecationWarning：警告提示一些将被Python未来版本淘汰的、弃用的部分。
• ImportWarning：当与 Import 语句相关的警告发生时，会引发 ImportWarning。
• ResourceWarning ：警告提示您可能正在泄漏资源，如文件句柄、网络连接等。

二、Python捕获警告的三种方法

1. 使用 warnings 模块

Python 中有一个 warnings 模块，可以用于捕获警告，通过该模块可以定义被触发时所应采取的行为。其使用方法如下：

import warnings
warnings.warn('警告的内容')

输出：

__main__:1: UserWarning: 警告的内容

可以发现在警告输出时，它会指定了问题所在的行数，毕竟这个警告是由我们主动打印的，并不是跑出来的。

2. 使用 warnings.catch_warnings() 上下文环境

import warnings

def catch_warning():
    with warnings.catch_warnings():
        warnings.simplefilter("always")
        warnings.warn("警告的内容", category=UserWarning)

catch_warning()

输出：

__main__:5: UserWarning: 警告的内容

使用上下文环境来捕获警告，会在运行期间临时启用警告处理程序。在该上下文环境中，只有指定警告类别的警告才会被触发。在上面的示例中，我们使用了 simplefilter() 函数来指定要捕获的警告类别。

3. 使用 warnings.filterwarnings() 过滤警告

import warnings

def filter_warning():
    warnings.filterwarnings("ignore", category=UserWarning)
    warnings.warn("警告的内容", UserWarning)

filter_warning()

通过 filterwarnings() 函数的调用，我们可以指定所需的警告选项，将选择存储在警告选项存储器中。在该示例中，我们使用 filterwarnings() 函数将 UserWarning 类别添加到被忽略的警告列表中。

三、两个示例说明

1. 使用 warnings 模块

import warnings
warnings.warn('Python 2.x 将不会为这些特殊序列行为产生 StopIteration 异常了。')

输出：

__main__:3: UserWarning: Python 2.x 将不会为这些特殊序列行为产生 StopIteration 异常了。

2. 使用 warnings.catch_warnings() 上下文环境

import warnings

def catch_warning():
    with warnings.catch_warnings():
        warnings.simplefilter('error', category=DeprecationWarning)
        print(type('hello'))

catch_warning()

输出：

<class 'str'>

这里没报错就是没抓到过期警告（DeprecationWarning），因为为捕获警告类型设置了错误级别，所以在上下文环境中只有指定警告类别的警告才会被触发。如果不设置 simplefilter() 函数，则 catch_warnings() 上下文环境将捕获所有警告。

以上就是 Python 捕获警告的三种方法，通过这三种方法，我们可以有效地捕获程序中出现的警告信息，进一步优化程序的可靠性和可读性。