我可以在列表理解的一部分中使用最终值吗？

Question

假设我有一个从 1 到 10 的数字列表。我想知道这些小于 1000 的数字的平方，而不关心实际数字。这是我正在尝试实现的一个简化示例，我想知道单个列表理解是否可以做到这一点。

我可以这样做：

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
squares_under_1000 = [n ** 2 for n in numbers if n ** 2 < 1000]
print(squares_under_1000) # Output: [1, 4, 27, 256]

问题是我需要使用两次实际的平方计算 - 我需要重复自己，如果操作很慢，那就是......慢。

我知道我也可以使用filter：

squares_under_1000 = list(filter(lambda sq: sq < 1000, [n ** 2 for n in numbers]))

但我的意思是不要以另一种方式这样做，因为我已经知道该怎么做。相反：是否可以使用列表推导来计算一个值并在我的推导的if 部分中使用该计算值？

Answer 1

这是在 Python 3.8 中添加赋值表达式的基本原理之一：

>>> [ n2 for n in numbers if (n2 := n ** n) < 1000 ]
[1, 4, 27, 256]

Answer 2

您的帖子要求输入小于 1000 的数字的平方，但您正在提高自己的数字；也就是说，x**x == x^x。我假设你想要 x^2 == x**2。

@chepner 对 Python 3.8 及更高版本有一个很好的答案。对于低于 python 3.8（或者如果你不喜欢海象运算符），嵌套列表理解可能是这里的票：

numbers = list(range(1,11))
sq_und_1000 = [v for v in [n**2 for n in numbers] if v < 1000]

Answer 3

查一下。 （在字典里。）

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
squares = {n: n**2 for n in numbers}
[v for v in squares.values() if v < 1000]

编辑：也可以在第二行创建另一个列表理解，它会满足您的要求。但是如果您还需要原始数字，那么字典会更好。

附：你在原来的帖子中提到了 n。