import warnings

import warnings 是 Python 中用于处理警告信息的模块。在编写代码时，有时会遇到一些情况，虽然不会导致程序直接报错，但可能会引发潜在的问题，比如使用了过时的函数、可能的类型错误等。warnings 模块允许开发者捕获和过滤这些警告信息，以便更好地控制程序的行为。

常见用法

1. 显示警告信息

默认情况下，Python 会显示警告信息，但不会中断程序运行。

import warnings

def deprecated_function():
    warnings.warn("This function is deprecated and will be removed in future versions.", DeprecationWarning)
    print("Function executed.")

deprecated_function()

2. 过滤警告信息

可以使用 warnings.filterwarnings 来控制警告信息的显示。

import warnings

# 忽略所有 DeprecationWarning
warnings.filterwarnings("ignore", category=DeprecationWarning)

def deprecated_function():
    warnings.warn("This function is deprecated and will be removed in future versions.", DeprecationWarning)
    print("Function executed.")

deprecated_function()

3. 将警告转换为异常

有时，为了确保代码的正确性，可以将某些警告转换为异常，这样当警告发生时程序会直接报错。

import warnings

# 将 DeprecationWarning 转换为异常
warnings.filterwarnings("error", category=DeprecationWarning)

def deprecated_function():
    warnings.warn("This function is deprecated and will be removed in future versions.", DeprecationWarning)
    print("Function executed.")

try:
    deprecated_function()
except DeprecationWarning as e:
    print(f"Caught warning as exception: {e}")

4. 暂时忽略警告

在某些情况下，可能只想暂时忽略警告信息。

import warnings

def deprecated_function():
    warnings.warn("This function is deprecated and will be removed in future versions.", DeprecationWarning)
    print("Function executed.")

with warnings.catch_warnings():
    warnings.simplefilter("ignore")
    deprecated_function()

常见警告类型

• UserWarning：通用警告，用于提醒用户注意某些情况。
• DeprecationWarning：用于标记即将废弃的功能。
• FutureWarning：用于标记未来版本中可能会改变行为的功能。
• RuntimeWarning：用于标记运行时可能出现的问题，如溢出等。

实际应用

在数据分析和机器学习中，经常会遇到各种警告信息，比如数据类型转换警告、收敛警告等。通过合理使用 warnings 模块，可以更好地管理这些警告信息，提高代码的可读性和可维护性。

import pandas as pd
import warnings

# 忽略特定的警告
warnings.filterwarnings("ignore", message="A value is trying to be set on a copy of a DataFrame")

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df['C'] = df['A'] + df['B']

通过 warnings 模块，可以有效地控制警告信息的显示，使代码更加简洁和高效。