Python;异常感知 map()

Question

我正在对 通用数据 进行一些 pyplotting 并将其从功率值转换为 dB 值。由于这些值来自系统，0 被用作“有用数据到此结束”的指示符（数学的性质，不是定义的值）。

我通常处理这些问题的方法是将转换包装在 try/except 中并返回默认的“低”值，例如

def f(value):
    try:
        return convert(value)
    except ValueError:
        return -140 #implementation specific, don't worry

这对于我的框架中 90% 的使用来说都很好，除了在绘图方面。

我很懒所以我现在要做的是：

pl.plot(xvals,map(f,yvals))

这会正确绘制数据，当数据结束时，会从悬崖上掉下来，这是预期的行为。但我希望图表在遇到 ValueError 异常时 end 并完全取消 f()。

除了将地图分解成一个循环之外，还有人有什么绝妙的想法吗？

更新：

我正在使用 Pylab / MatplotLib “端点”取决于执行；有时上述内容根本不重要，因为没有“坏”值。这一切都是为了让我变得懒惰并使用matplotlibs图形缩放而不是根据ydata的最小值重置动态ylim（我不做atm，在这种情况下只是ylim（-140）。）

关于答案的模糊重要更新： unutbu 的答案是我实际上将用于我的实现的，因为（在问题依赖项中没有提到），因为在这个经常使用的函数中引发 StopIteration 会对与问题无关的控制逻辑造成严重破坏，而不是把所有try-excepts 中的其他实例；有时 -inf 比你想象的更有意义。

感谢大家的速度非常快，我为 QuestionFail 向 unutbu 道歉。

Answer 1

如果您使用的是matplotlib，则表示您已安装numpy。

由于您要转换为dB，听起来您可能正在记录日志。在这种情况下，np.log(0) = -inf。

您可以使用 numpy 函数 np.ma.masked_invalid 屏蔽 nans 和 infs，matplotlib 可以绘制屏蔽数组。例如，

import matplotlib.pyplot as plt
import numpy as np

xvals=np.arange(100)
yvals=np.cumsum(np.random.random(100))
yvals[-10:]=0
yvals=np.log(yvals)

print(yvals[-10:])
# [-Inf -Inf -Inf -Inf -Inf -Inf -Inf -Inf -Inf -Inf]

yvals=np.ma.masked_invalid(yvals)
plt.plot(xvals,yvals)
plt.show()

产量

请注意，图表以 xval 结尾，等于 89，因为 yval 的最后 10 个值被屏蔽了。

Answer 2

拒绝使用循环结构是在不必要地限制自己。

在您的情况下，您希望在达到某个值时停止迭代数据，这正是 forloops 和 breaks 的目的

yvals_ = []
for y in yvals:
    y_ = f(y)
    if y_ == -140:
        break
    else:
        yvals_.append(y_)

p1.plot(xvals[:len(yvals_)],yvals_)

Answer 3

也许绘图库中有一些技巧，但更好的选择似乎一开始就不会生成此类数据。不是map为你省了三十行代码……

使用itertools.takewhile(lambda y: y != NO_VALUE, (f(y) for y in yvals))（如果绘图库需要列表而不是可迭代对象，则将其封装在对list 的调用中）。

编辑：我有一个更好的主意：在包装器中，添加

except ValueError:
    raise StopIteration

这是表示“迭代结束”的异常信号，map 尊重它。

Answer 4

您似乎有数据，并且您不想绘制最后一点。那么不绘制呢？

pl.plot(xvals[:-1], map(f, yvals)[:-1])