Question

我试图从列表列表中创建2个列表，下面的代码是一个简单易用的示例，但是认为有一个命令可以代替创建for循环进行拆分是有意义的。

def add_five(x):
    return x,x + 5
result1=[]
result2=[]
nums = [11, 22, 33, 44, 55]
result = list(map(add_five, nums))
print(result)
for n in range(len(result)):
    result1.append(result[n][0])
    result2.append(result[n][1])
print(result1,result2)

列表列表为：

[(11, 16), (22, 27), (33, 38), (44, 49), (55, 60)]
result1=[11, 22, 33, 44, 55] 
result2=[16, 27, 38, 49, 60]

有没有可以使用for循环帮助的命令？

Answer 1

您可以使用列表推导来提取每个tuple的 n 个元素：

L = [(11, 16), (22, 27), (33, 38), (44, 49), (55, 60)]

result1 = [i[0] for i in L]
result2 = [i[1] for i in L]

功能上，您可以将operator.itemgetter用作等效项：

from operator import itemgetter

result1 = list(map(itemgetter(0), L))
result2 = list(map(itemgetter(1), L))

如果您不知道元组的大小，建议的解决方案是zip，它将输出一个元组列表，每个元组代表一个元组索引：< / p>

results = list(zip(*L))

[(11, 22, 33, 44, 55), (16, 27, 38, 49, 60)]

您甚至可以在此处使用序列解压缩而无需形成完整列表：

result1, result2 = zip(*L)

用于将列表列表拆分为每个列表的第一个元素的列表的命令

1 个答案: