从警告回溯

Question

我有一个代码，在某些时候会显示警告，我认为它在计算 mean() 时遇到问题

我想知道是否有任何方法可以强制 python 告诉我在哪里，或者哪一行，或者比这个消息更多的信息：

C:\Python27\lib\site-packages\numpy\core\_methods.py:55: RuntimeWarning: Mean of empty slice.
  warnings.warn("Mean of empty slice.", RuntimeWarning)
C:\Python27\lib\site-packages\numpy\core\_methods.py:79: RuntimeWarning: Degrees of freedom <= 0 for slice
  warnings.warn("Degrees of freedom <= 0 for slice", RuntimeWarning)

我不知道是否可以“捕获”警告..​​...如果我有任何错误，通常我使用的是回溯包：

import traceback

然后我通常会这样做：

try:
    #something
except:
    print traceback.format_exc()

Answer 1

您可以将警告变成异常：

import warnings

warnings.simplefilter("error")

现在不是打印警告，而是引发异常，为您提供回溯。

-W command line switch可以达到同样的效果：

$ python -W error somescript.py

或通过设置PYTHONWARNINGS environment variable:

$ export PYTHONWARNINGS=error

您可以与其他warnings.simplefilter() arguments 一起玩，以更具体地了解应该引发异常的警告。例如，您可以过滤 warnings.RuntimeWarning 和行号。