Skip to main content
 首页 » 编程设计

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

2025年05月04日16wayfarer

我有这样的数据:

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] 

请您参考如下方法:

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

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

假设 col1 和 col2 的长度相等