如果函数在python中在调用者级别引发警告，则打印消息

Question

我有一个在我的代码中多次调用的函数f。此函数的详细信息并不重要，除了它有时可能生成RuntimeWarning，因为函数内的迭代没有进展。当发生这种情况时，我希望向用户打印一条消息，警告他们哪个输入导致该功能发出警告。示例：

for x in numpy.arange(5):
    y = f(x)
    #if calculation of y raised a RuntimeWarning then print "x = %.2f is problematic" % x

任何解决方案？

编辑：

我已经看到引用的问题是这篇文章重复的原因，但我觉得我的问题不同。如果为特定函数调用生成警告，我只想打印一条消息...不引发任何异常。