如何在DataFrame条件判断中添加错误/警告提示？

首先构造示例DataFrame：

import pandas as pd

df = pd.DataFrame({
    'trade': ['1', '1', '3', '3'],
    'venue': ['TSE', 'LSE', 'ABC', 'TTT']
})

方法1：用warnings模块发出警告（不终止代码） #

先筛选出trade == '3'且venue不在['ABC', 'BBB']的行，若存在这类行则触发警告：

import warnings

# 筛选违规行
invalid_rows = df[(df['trade'] == '3') & (~df['venue'].isin(['ABC', 'BBB']))]

if not invalid_rows.empty:
    warning_content = f"发现不符合要求的记录：\n{invalid_rows.to_string(index=False)}"
    warnings.warn(warning_content)

这种方式会以标准警告格式输出信息，程序会继续执行。

方法2：抛出错误（终止代码） #

如果需要强制终止程序并提示错误，可使用raise抛出异常：

invalid_rows = df[(df['trade'] == '3') & (~df['venue'].isin(['ABC', 'BBB']))]

if not invalid_rows.empty:
    error_content = f"存在违规记录：\n{invalid_rows.to_string(index=False)}"
    raise ValueError(error_content)

触发后程序会停止运行，并在控制台输出错误信息。

方法3：直接打印提示信息（轻便无依赖） #

不需要额外模块，直接自定义打印提示：

invalid_rows = df[(df['trade'] == '3') & (~df['venue'].isin(['ABC', 'BBB']))]

if not invalid_rows.empty:
    print("⚠️ 警告：发现不符合要求的记录：")
    print(invalid_rows.to_string(index=False))

适合快速调试或简单场景，程序不会中断。

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