数组上Lambda表达的模棱两可的评估

Question

我想使用一个lambda ，如果x等于零，则将其加到x。 我尝试了以下表达式：

t = map(lambda x: x+1 if x==0 else x, numpy.array())
t = map(lambda x: x==0 and x+1 or x, numpy.array())
t = numpy.apply_along_axis(lambda x: x+1 if x==0 else x, 0, numpy.array())

这些表达式中的每一个都返回以下错误：

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我对map()和numpy.apply_along_axis()理解是，它将采用一些函数并将其应用于数组的每个值。 从错误看来，lambda被评估为x=array ，而不是x=array某个值。 我究竟做错了什么？

我知道我可以编写一个函数来完成此操作，但是我想对python的功能编程方面更加熟悉。

Answer 1

如果您使用的是numpy，则应编写矢量化代码：

arr + (arr == 0)