python - 如何根据另一个计数列表复制列表元素？

Question

我有两个列表，我正在尝试根据它们出现的次数创建多个数字副本。

numbers = [0, 1, 2]
amount = [1, 2, 3]

我试过了：

total = []
n = 0
for i in range(len(numbers)):
    product = numbers[n] * amount[n]
    n += 1
    total.extend(product)

但我得到了错误：

TypeError: 'int' object is not iterable.

我的预期输出是：

total = [0, 1, 1, 2, 2, 2]

score 3 · Accepted Answer

利用zip

result = []
for i, j in zip(numbers, amount):
    result.extend([i] * j)

print(result)

[0, 1, 1, 2, 2, 2]

score 2 · Accepted Answer

您的错误在以下行内：

product = numbers[n] * amount[n]

它不会产生一个列表，而是一个整数。因为你将两个数字相乘。

你真正想要的是

product = [numbers[n]] * amount[n]

2 回答 2