将函数应用于可返回多行的pandas DataFrame

Question

我正在尝试转换DataFrame，以便将某些行复制给定的次数。例如：

df = pd.DataFrame({'class': ['A', 'B', 'C'], 'count':[1,0,2]})

  class  count
0     A      1
1     B      0
2     C      2

应该转换为：

  class 
0     A   
1     C   
2     C

这与使用count函数进行聚合相反。有没有一种简单的方法可以在熊猫中实现（不使用for循环或列表推导）？

一种可能是允许DataFrame.applymap函数返回多行（的类似apply方法GroupBy）。但是，我认为现在在大熊猫中是不可能的。

Answer 1

您可以使用groupby：

def f(group):
    row = group.irow(0)
    return DataFrame({'class': [row['class']] * row['count']})
df.groupby('class', group_keys=False).apply(f)

所以你得到

In [25]: df.groupby('class', group_keys=False).apply(f)
Out[25]: 
  class
0     A
0     C
1     C

您可以根据需要固定结果的索引