使用列表理解:
output_age = [i for l in ([a]*f for a, f in zip(age, frequency)) for i in l]
#[19, 19, 20, 21, 22, 22, 22, 23, 23, 24, 25]
为什么?
我们首先将zip 放在一起age 和frequency 列表,以便我们可以一致地迭代它们。因此:
for a, f in zip(age, frequency):
print(a, f)
给予:
19 2
20 1
21 1
22 3
23 2
24 1
25 1
然后我们要重复每个元素a,重复次数与f 确定的一样多。这可以通过创建一个列表并将其相乘来完成。就像:
[4] * 3
#[4, 4, 4]
然后我们需要解压这些值,因此我们将这个表达式包装在一个生成器中(用括号表示)并对其进行迭代。这会使列表变平。注意有alternative ways实现这个(比如使用itertools.chain.from_iterable)。
另一种方法是重复数字 a,通过迭代 range 对象而不是乘以列表来获得重复次数。
这个方法看起来像:
output_age = [a for a, f in zip(age, frequency) for _ in range(f)]
#[19, 19, 20, 21, 22, 22, 22, 23, 23, 24, 25]