python - 将 lambda 表达式应用于数组元素时出现 ValueError

Question

目前我在处理 numpy.array - 4x1 - 即时遇到错误

[[-1.96113883]
 [-3.46144244]
 [ 5.075857  ]
 [ 1.77550086]]

使用 lambda 函数 f = lambda x: x if (x > 0) else (x * 0.01)。

错误是 ValueError: 具有多个元素的数组的真值不明确。使用 a.any() 或 a.all()。

我在 stackoverflow.com 上搜索了不同的主题，但没有找到任何令人满意的问题解释和适合我的案例(许多对 和 运算符的不明确引用，矢量化代码等.).

在处理完数组后，我期望的是一个与输入数组具有相同维度的数组，并且每个值都根据函数进行了修改，例如:

[[-0.0196113883]
 [-0.0346144244]
 [ 5.075857  ]
 [ 1.77550086]]

最后，有人能给我一个解决方案和解释为什么会出现这个错误吗？谢谢你的建议。

Answer 1

x > 0 作为一个整体对您的 numpy 数组进行评估，返回另一个 bool 数组。但是，if 语句将整个数组计算为单个操作。

arr = np.array([[-1.96113883],
                [-3.46144244],
                [ 5.075857  ],
                [ 1.77550086]])
print arr > 0
   [[False]
    [False]
    [ True]
    [ True]]

如错误消息中所述， bool 数组的真值不明确。

相反，正如 ajcr 在评论中指出的那样，您应该使用 np.where 作为矢量化 if-else 语句

例如

np.where(arr > 0, arr, arr*0.01)
array([[-0.01961139],
       [-0.03461442],
       [ 5.075857  ],
       [ 1.77550086]])