Python 熊猫基于以前的值创建组

我有一个按整数列排序的数据帧

v1

：

   v1
0   1
1   5
2   6
3  12
4  15
5  23
6  24
7  25
8  33

我想对v1中的值进行如下分组：如果值-prev_值<5，则它们具有相同的组。
为此，我想为每个小组提供一个越来越多的数字

因此，我想创建另一个列，

v1\u group

，它将有以下输出：

   v1  v1_group
0   1         1 
1   5         1  
2   6         1
3  12         2  # 12 - 6 > 5, new group
4  15         2
5  23         3
6  24         3
7  25         3
8  33         4

我需要对datetime列执行相同的任务：如果value-prev_value 我知道我可以用标准for循环解决这个问题。有更好的方法吗？

IIUC

df['v1_group'] = df.v1.diff().ge(5).cumsum() + 1

输出：

   v1  v1_group
0   1         1
1   5         1
2   6         1
3  12         2
4  15         2
5  23         3
6  24         3
7  25         3
8  33         4