我有一个返回元组的函数 foo()。比方说 ('a', 1) 我有一个可迭代的 a ,我想对其进行迭代并将每个项目传递给该函数。 最后,我需要两个列表 - r1 和 r2,其中 r1 包含从函数返回的元组中的所有第一项。第二个列表 r2 - 这些元组中所有第二项的结果。
r1 = ['a', 'b', 'c']
r2 = [1, 2, 3]
我不太喜欢这种方法:
result = [foo(i) for i in a]
r1 = [i[0] for i in result]
r2 = [i[1] for i in result]
请您参考如下方法:
您可以使用 zip功能。
>>> result = [('a', 1), ('b', 2), ('c', 3)]
>>> r1, r2 = zip(*result)
>>> r1
('a', 'b', 'c')
>>> r2
(1, 2, 3)
zip(*result) 解压 result 中的每个列表并将它们作为单独的参数传递到 zip功能。它本质上是transposes列表。它生成一个包含两个元组的列表,然后将其分配给 r1 和 r2。
