python - 是否可以在 lambda 函数中包含多个方程

Question

我试图在 lambda 函数中包含具有不同长度的变量的多个操作，即:

$ serial_result = map(lambda x,y:(x**2,y**3), range(20), range(10))

但这不起作用。有人可以告诉我如何解决这个问题吗？ 我的理解是:

$ serial_result = map(lambda x,y:(x**2,y**3), range(0,20,2), range(10))

之所以有效，是因为“x”和“y”的数组具有相同的长度。

Answer 1

如果您想要范围项目的产品，您可以使用itertools.product:

>>> from itertools import product
>>> serial_result = map(lambda x:(x[0]**2,x[1]**3), product(range(20), range(10)))

如果你想像第二种情况一样将这些对传递给 lambda，你可以使用 itertools.zip_longest(在 python 2 中使用 izip_longest)并传递一个 fillvalue 来填充缺失的值项目，

>>> from itertools import zip_longest
>>> serial_result = map(lambda x:(x[0]**2,x[1]**3), zip_longest(range(20), range(10),fillvalue=1))

请注意，如果您使用的是 python 2，您可以将多个参数作为元组传递给 lambda :

>>> serial_result = map(lambda (x,y):(x**2,y**3), product(range(20), range(10)))

在以下示例中查看 izip_longest 和 product 的区别:

>>> list(product(range(5),range(3)))
[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2), (3, 0), (3, 1), (3, 2), (4, 0), (4, 1), (4, 2)]
>>> list(zip_longest(range(5),range(3)))
[(0, 0), (1, 1), (2, 2), (3, None), (4, None)]
>>> list(zip_longest(range(5),range(3),fillvalue=1))
[(0, 0), (1, 1), (2, 2), (3, 1), (4, 1)]