Aplicar função ao pandas DataFrame que pode retornar várias linhas
Eu estou tentando transformar DataFrame, de tal forma que algumas das linhas serão replicadas um determinado número de vezes. Por exemplo:
df = pd.DataFrame({'class': ['A', 'B', 'C'], 'count':[1,0,2]})
class count
0 A 1
1 B 0
2 C 2
deve ser transformado para:
class
0 A
1 C
2 C
Este é o contrário da agregação com a função count. Existe uma maneira fácil de alcançá-lo em pandas (sem usar loops ou compreensão de listas)?
Uma possibilidade poderia ser permitirDataFrame.applymap
função retornar várias linhas (semelhanteapply
método deGroupBy
). No entanto, não creio que seja possível nos pandas agora.