在lambda功能中弄平列表

Question

我正在尝试构建一个lambda表达式，该表达式以：（a，b []，c），其中b是需要的元素列表被扁平。

例如，表达式的输出应为： （a，b [0]，c），（a，b [1]，c），（a，b [2]，c）...

我尝试研究并寻找与我刚刚描述的情况相似的发电机表达方式，但我无法真正缠绕着这个。

如果有人可以将我指向正确的方向，我会很感激。

Answer 1

您可以使用 itertools.repeats.repeats.repeat（） /a>使用zip（）为b中的每个元素创建所需的元组：

from itertools import repeat

f = lambda a, b, c: zip(repeat(a), b, repeat(c))

print(*f(1, [2, 3, 4], 5))

此输出：

(1, 2, 5) (1, 3, 5) (1, 4, 5)

Answer 2

返回列表的lambda函数理解列表长度的参数。

func = lambda a, b, c: [(a, b[i], c) for i in range(len(b))]

print(func(1, [2, 3, 4], 5))