python忽略warnings的几种方法_Python

我在训练深度学习模型的时候一直出现这样的警告，但是不影响运行：

userwarning: failed to load image python extension: [winerror 127] 找不到指定的程序。
warn(f"failed to load image python extension: {e}")

要避免在 python 程序运行时显示类似的警告信息，可以使用 python 内置的警告控制机制来抑制特定类型的警告。可以通过以下几种方法来抑制这类警告。

方法 1: 使用 warnings 模块过滤特定类型和消息内容的警告

python 的 warnings 模块允许控制警告的显示。在代码中添加以下内容，以抑制 userwarning：

import warnings

# 忽略指定的警告类型
warnings.filterwarnings("ignore", category=userwarning, message="failed to load image python extension")

# 其他代码...

这样，只有与该特定消息匹配的 userwarning 会被忽略，其他警告仍然会显示。

方法 2: 使用 warnings 模块过滤特定类型的警告

抑制所有的 userwarning类型的警告，可以使用以下代码：

import warnings

# 忽略所有 userwarning 类型的警告
warnings.filterwarnings("ignore", category=userwarning)

# 其他代码...

这样，只有 userwarning 类型的警告会被忽略，其他警告仍然会显示。

方法 3: 使用 pythonwarnings 环境变量

如果不想修改代码，可以通过设置环境变量 pythonwarnings 来抑制特定类型的警告。例如，在命令行或脚本运行时可以这样设置：

linux/macos
pythonwarnings=ignore::userwarning python your_script.py
windows
set pythonwarnings=ignore::userwarning
python your_script.py

这种方式会全局抑制指定的警告类型，而不需要修改代码。

方法 4: 临时抑制所有警告（不推荐，但不显示任何警告信息）

如果想完全抑制所有警告，可以使用以下代码：

import warnings

# 忽略所有警告
warnings.simplefilter("ignore")

# 其他代码...

注意：完全忽略所有警告，因为警告信息通常能帮助发现潜在的问题。最好是只抑制那些明确知道不会影响程序运行的警告。

方法 5: 控制过滤警告输出的其他设置方法：

import warnings

warnings.simplefilter("default")  # 这是默认的警告过滤器设置，它会显示所有警告信息。
warnings.simplefilter("error")  # 这个设置会将所有警告转换为错误，这意味着如果代码触发了警告，python解释器会抛出一个异常并停止执行。
warnings.simplefilter("always")  # 无论警告是在哪里产生的，这个设置都会显示警告信息。
warnings.simplefilter("module")  # 这个设置会在每个模块的第一条警告信息被显示后，忽略该模块后续的所有警告。
warnings.simplefilter("once")  # 这个设置会确保每个警告信息只被显示一次，即使在代码中多次触发了相同的警告。

方法 6: 通过日志库控制警告输出

如果使用日志库（如 logging）来管理程序的日志输出，可以将警告信息重定向到日志系统，并根据需要决定是否记录或丢弃这些警告。以下是一个简单示例：

import logging
import warnings

# 设置日志记录器
logging.capturewarnings(true)
logger = logging.getlogger("my_logger")
logger.setlevel(logging.error)  # 只记录错误级别以上的日志

# 创建一个处理器，比如输出到文件
file_handler = logging.filehandler("my_log_file.log")
file_handler.setlevel(logging.error)
logger.addhandler(file_handler)

# 你的其他代码...

# 这将会被记录到日志文件中，而不会在控制台显示
warnings.warn("this warning will be logged, but not sihown in the console.")