python - 反向分组

Question

我有一个带有变量名称的 Pandas 数据框，每个变量的值和 count (显示该行的频率):

df = pd.DataFrame({'var':['A', 'B', 'C'], 'value':[10, 20, 30], 'count':[1,2,3]})

var  value  count
A    10     1
B    20     2
C    30     3

我想用 count得到这样的输出:

var  value
A    10
B    20
B    20
C    30
C    30
C    30

最好的方法是什么？

Answer 1

您可以使用 index.repeat :

i = df.index.repeat(df['count'])
d = df.loc[i, :'value'].reset_index(drop=True)

   var  value
0   A     10
1   B     20
2   B     20
3   C     30
4   C     30
5   C     30