python – 如何基于计数器列复制行

前端之家收集整理的这篇文章主要介绍了python – 如何基于计数器列复制行前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
假设我有一个名为df的数据框
x count 
d 2
e 3
f 2

计数将是计数器列和#次我希望它重复.

我将如何扩展它来实现它

x count
d 2
d 2
e 3
e 3
e 3
f 2
f 2

我已经试过了
numpy.repeat(df,df.iloc [‘count’])并且它出错了

解决方法

你可以使用np.repeat()
import pandas as pd
import numpy as np

# your data
# ========================
df

   x  count
0  d      2
1  e      3
2  f      2

# processing
# ==================================
np.repeat(df.values,df['count'].values,axis=0)


array([['d',2],['d',['e',3],['f',2]],dtype=object)

pd.DataFrame(np.repeat(df.values,axis=0),columns=['x','count'])

   x count
0  d     2
1  d     2
2  e     3
3  e     3
4  e     3
5  f     2
6  f     2

猜你在找的Python相关文章