python - 根据值范围将数据帧拆分为较小的数据帧

Question

我有以下数据框:

     x     text     
1    500   aa
2    550   bb
3    700   cc
4    750   dd 

如果 x 值相距超过 100 点，我的目标是拆分此 df。

是否有一个 Pandas 函数可以让您根据值的范围进行拆分？

这是我想要的输出:

df_1:

    x     text  
0   500   aa
1   550   bb

df_2:

    x     text  
0   700   cc
1   750   dd

Answer 1

我相信您需要通过助手将 groupby 对象转换为元组和字典 Series :

d = dict(tuple(df.groupby(df['x'].diff().gt(100).cumsum())))
print (d)
{0:      x text
1  500   aa
2  550   bb, 1:      x text
3  700   cc
4  750   dd}

详情 :

先通过 Series.diff 得到差值, 比较 Series.gt 更大并通过 Series.cumsum 创建连续组:

print (df['x'].diff().gt(100).cumsum())
1    0
2    0
3    1
4    1
Name: x, dtype: int32