如何让Python脚本触发RuntimeWarning时中止以定位警告位置？

嘿，这个场景太熟悉了！用tqdm把循环速度拉起来后，警告消息一闪而过，根本抓不住是哪次循环出的问题。别着急，我们可以把Python的警告强制转换成错误，这样只要触发那个除以零的警告，脚本就会立刻停止，还能给出完整的报错栈，帮你精准定位问题。

下面给你几个实用的方案：

1. 用warnings模块把指定警告转为错误 #

这是最直接的方法，你可以在脚本开头添加几行代码，让Python把"divide by zero encountered in true_divide"这个警告当成错误抛出：

import warnings

# 精准匹配除以零的警告，将其转为错误
warnings.filterwarnings("error", message="divide by zero encountered in true_divide")

如果想对所有RuntimeWarning都生效（比如还有其他可能的运行时警告需要排查），可以写成：

warnings.filterwarnings("error", category=RuntimeWarning)

这样一来，当循环里触发除以零的操作时，脚本会立刻停止，控制台会打印出完整的报错追踪信息，告诉你具体是哪一行代码、第几次循环出的问题。

2. 结合Spyder的调试功能深挖问题 #

当脚本因为警告转错误停止后，你可以在Spyder的IPython控制台里输入%debug魔法命令，直接进入调试模式。在这个模式下，你可以查看当时的变量值（比如循环的索引、当前处理的数值），甚至一步步回溯代码执行过程，彻底搞清楚为什么会出现除以零的情况。

3. 临时关闭tqdm（备选方案） #

如果你不想修改警告处理逻辑，也可以临时注释掉tqdm的装饰器，让循环慢下来，这样就能看到警告触发时的上下文了。不过你提到这种方法执行太慢，所以优先用前两种方案。

示例代码 #

给你写个简单的例子，直观看看效果：

import warnings
import numpy as np
from tqdm import tqdm

# 开启警告转错误
warnings.filterwarnings("error", message="divide by zero encountered in true_divide")

# 模拟你的循环数据
sample_data = [2, 5, 0, 3, 0]

for idx, num in enumerate(tqdm(sample_data)):
    print(f"当前循环索引: {idx}")
    # 这里会触发除以零的警告/错误
    result = np.true_divide(10, num)

运行这段代码，当循环到num=0时，脚本会立刻停止，你能看到清晰的报错信息，直接定位到出问题的那一行。

排查完问题后，记得把警告处理改回默认状态，避免影响其他正常警告：

warnings.filterwarnings("default")

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