`warnings.warn`跨环境行为不一致及相关技术问题咨询

测试代码如下：

def f():
    import warnings
    warnings.warn("deprecated", DeprecationWarning, stacklevel=2)

def g():
    return f()

def h():
    return g()

h() # or f() or g()

问题1：Colab与终端的警告表现差异原因 #

终端中h()始终无警告的原因 #

是的，这完全由stacklevel=2导致。Python的警告系统默认会去重相同的警告（判断依据是警告消息、类别、触发的文件和行号）：

• 调用f()时，stacklevel=2让警告指向直接调用f()的代码行；
• 调用g()时，警告指向g()内部调用f()的行（return f()）；
• 调用h()时，h()调用g()，g()再调用f()，stacklevel=2会跳过f()的栈帧，指向调用f()的位置——也就是g()里的return f()行，和调用g()时的警告触发位置完全一致，因此被判定为重复警告直接忽略。

Colab中的表现差异 #

Colab的单元格执行环境有特殊的上下文缓存：

• 单独运行h()时，当前单元格没有触发过来自g()内部调用f()的警告，警告系统无对应去重记录，所以会显示这条警告；
• 同单元格依次运行f()、g()、h()时，g()已经触发过来自g()内部return f()行的警告，h()触发的警告和这条完全重复，因此被去重，最终只显示2条。

Colab与终端的核心差异 #

两者的警告去重逻辑一致，但Colab的单元格执行是交互式独立上下文：单独运行单元格时，警告系统的去重缓存是全新的；而终端中连续执行代码时，去重缓存会持续保留至会话结束。

问题2：实现f()每次调用都显示警告 #

要让每次调用f()都触发警告，核心是绕过警告系统的去重机制，常用两种方法：

方法1：关闭警告去重 #

通过warnings.simplefilter设置警告始终显示，不管是否重复：

import warnings
# 对DeprecationWarning关闭去重，强制显示所有实例
warnings.simplefilter("always", DeprecationWarning)

def f():
    warnings.warn("deprecated", DeprecationWarning, stacklevel=2)

def g():
    return f()

def h():
    return g()

f()
g()
h()

执行后会显示3条独立警告，分别对应f()、g()、h()的调用位置。

方法2：动态调整stacklevel #

让每次警告的触发位置指向最终调用者的代码行，避免重复：

import inspect
import warnings

def f():
    # 计算当前栈深度，让警告指向最外层调用者的代码行
    stack_level = len(inspect.stack()) - 1
    warnings.warn("deprecated", DeprecationWarning, stacklevel=stack_level)

def g():
    return f()

def h():
    return g()

f()
g()
h()

这种方法会让警告分别指向f()、g()、h()的调用行，每个位置唯一，因此不会被去重。

问题3：stacklevel和skip_file_prefixes的作用 #

stacklevel #

用于指定警告消息指向的栈帧层级：

• stacklevel=1：警告指向warnings.warn所在的代码行（即f()内部的警告语句）；
• stacklevel=2：警告指向调用当前函数的代码行（比如g()里调用f()的行，或直接调用f()的行）；
• 数值越大，指向的栈帧越外层（越接近最终用户的调用代码）。
  在你的场景中，stacklevel=2的作用是让警告提示用户“你调用的某个函数已废弃”，而非指向函数内部的警告语句。

skip_file_prefixes #

这是Python警告系统的内部配置，用于过滤系统级文件的栈帧（如Python标准库、Colab运行时内部文件），确保警告消息仅显示用户编写的代码位置。它会自动跳过路径前缀匹配的文件，避免警告指向用户无法修改的内部代码。

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