查找列表中值大于常量的第一个元素

Question

我有一个列表，我有一个常数，我想知道列表中的哪些元素大于常数。

ls = [2, 1, 0, 1, 3, 2, 1, 2, 1]
constant = 1.5

所以我只是做了：

ls >= constant

我希望它会返回：

[TRUE, FALSE, FALSE, FALSE, TRUE, TRUE, FALSE, TRUE, FALSE]

但是它返回了一个错误！！！

TypeError                     Traceback (most recent call last)
<ipython-input-92-91099ebca512> in <module>
----> 1 ls > constant

TypeError: '>' not supported between instances of 'list' and 'float'

如何 python 将向量与简单的常数值进行比较？

Answer 1

您可以在循环时使用列表推导来评估值。

比较元素时需要循环，因为您无法将list与integer进行比较。

output = [x > constant for x in ls]
#[True, False, False, False, True, True, False, True, False]

Answer 2

您可以为此使用map()和 lambda 函数。

print(list(map(lambda n:n>=constant, ls)))

您的代码不起作用的原因是无法比较列表和浮点数。 为了进行比较，它们需要具有相同的 object（或者预定义的魔术方法以使用自定义对象）。

Answer 3

获取您描述的列表的最 Pythonic (IMO) 方法是使用列表理解：

>>> [n >= constant for n in ls]
[True, False, False, False, True, True, False, True, False]

它与您编写的内容几乎相同，但您希望对 ls 中的 n 的每个单独的n值for n in ls比较，而不是对ls本身进行比较。

如果您想要实际值，这是一个非常相似的表达式：

>>> [n for n in ls if n >= constant]
[2, 3, 2, 2]

在这种情况下，列表中的实际值为n ，但我们仅在if n >= constant时才包含它。