python - 如何对排序的 pandas.Series 进行分组？

Question

给定一个已排序的 pandas.Series(或只是一个列表)对象，我想创建组(例如，列表或 pandas.Series)，这样相邻的对象之间的差异组中的元素小于某个阈值，例如:

THRESHOLD = 2
sorted_list = [1, 2, 10, 15, 16, 17, 20, 21]
# ...
result = [[1, 2], [10], [15, 16, 17], [20, 21]]

Answer 1

可以使用diff和cumsum来标记分组，然后使用groupby:

s = pd.Series(sorted_list)
s.groupby(s.diff().gt(THRESHOLD).cumsum()).apply(list).tolist()
# [[1, 2], [10], [15, 16, 17], [20, 21]]