Python如何根据另一列表对应位置数值重复当前列表的元素

实现方法 #

你可以直接用列表推导式配合zip函数完成需求，这是最符合Python风格的写法：

list1 = ['Name1', 'Name2', 'Name3']
list2 = [1, 3, 2]

result = [item for item, count in zip(list1, list2) for _ in range(count)]
print(result)

运行后输出即为你需要的结果：

['Name1', 'Name2', 'Name2', 'Name2', 'Name3', 'Name3']

逻辑说明

• 用zip将两个列表的对应位置元素配对，得到('Name1',1)、('Name2',3)、('Name3',2)三组配对值
• 两层循环的列表推导式中，外层遍历每一组的元素和重复次数，内层按照次数重复添加元素到结果列表

如果你更偏好可读性更强的普通循环写法，也可以用以下方式：

result = []
for item, count in zip(list1, list2):
    # 生成重复count次的元素列表，追加到结果中
    result.extend([item] * count)

内容的提问来源于stack exchange，提问作者Scott Woodhall