python从两个列表中创建一个新列表，第二个列表表示重复第一个列表中的元素多少次

Question

我有这样的数据:

df
col1 col2
3      4
1      2
2      2
4      7

现在数据是 pandas df，但我可以想象列列表或任何需要的东西。

我希望输出是这样的。

[3,3,3,3,1,1,2,2,4,4,4,4,4,4,4]

Answer 1

如果 col1 和 col2 可以表示为列表，则:

ans = []
for i in xrange(len(col1)):
    ans+=[col1[i]]*col2[i]

print ans

假设 col1 和 col2 的长度相等