python - 如何使用 3.x 模拟 2.x 的 lambda 参数元组解包？

Question

在Python 2中，我可以写:

In [5]: points = [ (1,2), (2,3)]

In [6]: min(points, key=lambda (x, y): (x*x + y*y))
Out[6]: (1, 2)

但是 3.x 不支持:

  File "<stdin>", line 1
    min(points, key=lambda (x, y): (x*x + y*y))
                           ^
SyntaxError: invalid syntax

简单的解决方法是显式索引到传递的元组中:

>>> min(points, key=lambda p: p[0]*p[0] + p[1]*p[1])
(1, 2)

这太难看了。如果 lambda 是一个函数，我可以这样做

def some_name_to_think_of(p):
    x, y = p
    return x*x + y*y

但是由于 lambda 仅支持单个表达式，因此无法将 x, y = p 部分放入其中。

我还能如何解决此限制？

Answer 1

不，没有其他办法。你涵盖了这一切。解决方法是在Python ideas mailing list上提出这个问题。 ，但要准备好在那里进行大量争论以获得一些吸引力。

实际上，并不是说“没有出路”，第三种方法可能是实现多一级的 lambda 调用，只是为了展开参数 - 但这会比你的方法更加低效和难以阅读。两个建议:

min(points, key=lambda p: (lambda x,y: (x*x + y*y))(*p))

<小时/>

Python 3.8 更新

自 Python 3.8 发布以来，PEP 572(赋值表达式)已作为工具提供。

因此，如果使用一种技巧在 lambda 内执行多个表达式 - 我通常通过创建一个元组并仅返回它的最后一个组件来实现这一点，则可以执行以下操作:

>>> a = lambda p:(x:=p[0], y:=p[1], x ** 2 + y ** 2)[-1]
>>> a((3,4))
25

应该记住，这种代码很少比具有完整功能的代码更具可读性或实用性。尽管如此，还是有可能的用途 - 如果有各种单行代码可以在此点上进行操作，那么有一个namedtuple可能是值得的，并使用赋值表达式来有效地将传入序列“转换”到命名元组:

>>> from collections import namedtuple
>>> point = namedtuple("point", "x y")
>>> b = lambda s: (p:=point(*s), p.x ** 2 + p.y ** 2)[-1]