Python中定位RuntimeWarning触发位置（不中断循环）

你的代码中因为数组y包含0元素，执行除法时会触发除以0的RuntimeWarning。如果直接将警告转为错误会中断循环，以下两种方法可以帮你定位警告触发位置，同时保持循环继续执行：

方法一：循环内捕获并记录警告 #

这种方式可以精准对应到每次循环的触发情况，方便定位具体是哪一次循环出了问题：

import numpy as np
import warnings
import traceback

# 确保警告总是被捕获（不会被过滤）
warnings.simplefilter("always", RuntimeWarning)

x = np.array([4, 5, 5, 7, 0,10])
y = np.array([2, 4, 6, 7, 0,2])

for i in range(len(x)):
    tmp_x = x[i]
    tmp_y = y[i]
    # 捕获当前循环内的警告
    with warnings.catch_warnings(record=True) as w:
        result = np.divide(tmp_x, tmp_y)
        print(result)
        # 若有警告，打印详细信息和调用栈
        if w:
            print(f"=== 循环第{i+1}次触发警告 ===")
            for warning in w:
                print(f"警告类型: {warning.category.__name__}")
                print(f"警告内容: {warning.message}")
                print("触发位置调用栈:")
                traceback.print_tb(warning.stack[0])

方法二：全局自定义警告处理器 #

通过替换默认的警告处理器，全局监控所有RuntimeWarning，无需在循环内单独处理：

import numpy as np
import warnings
import traceback

def custom_warn_handler(message, category, filename, lineno, file=None, line=None):
    # 打印警告核心信息
    print(f"\n=== 触发警告 ===")
    print(f"类型: {category.__name__}")
    print(f"内容: {message}")
    print(f"触发位置: 文件 {filename}，第 {lineno} 行")
    # 打印调用栈，定位代码执行路径
    print("调用栈:")
    traceback.print_stack(limit=5)

# 替换默认警告处理函数
warnings.showwarning = custom_warn_handler

x = np.array([4, 5, 5, 7, 0,10])
y = np.array([2, 4, 6, 7, 0,2])

for i in range(len(x)):
    tmp_x = x[i]
    tmp_y = y[i]
    print(np.divide(tmp_x, tmp_y))

两种方法都不会中断循环，同时能清晰展示警告的触发位置、内容和调用栈，帮你快速定位问题代码段。

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