熊猫DataFrame条件分组依据

时间:2018-12-16 14:57:20

标签: python pandas pandas-groupby

我有这个DF:

df = pd.DataFrame(data=[[-2.000000, -1.958010,  0.2],
                   [-1.958010, -1.916030,  0.4],
                   [-1.916030, -1.874040,  0.3],  
                   [-1.874040, -1.832050,  0.6],
                   [-1.832050, -1.790070, 0.8],
                   [-1.790070, -1.748080,  0.2]],columns=['egystart','egyend','fx'])

所以我想每两行进行分组,并得到fx作为两行的平均值。 egystart应该以第一行的egystart开头,egyend应该以第二行的egyend开头。

在这种情况下,我应该获得:

-2.000000 -1.916030  0.3
-1.916030 -1.832050  0.45  
-1.832050 -1.748080  0.5

所以我尝试过这样的事情:

df.groupby((df.egystart == df.egyend.shift(1)).cumsum()).agg({'egystart':min, 'egyend':max, 'fx':HERE_THE_MEAN_VALUE})

但这不起作用

1 个答案:

答案 0 :(得分:1)

您可以尝试执行以下操作以获取每2行fx的平均值:

result = df.groupby(np.arange(len(df))//2).mean()

print(result)

   egystart    egyend    fx
0 -1.979005 -1.937020  0.30
1 -1.895035 -1.853045  0.45
2 -1.811060 -1.769075  0.50