Python - 按符号对连续元素进行分组

在处理一些数据的时候，我们需要按照一定的规则对连续的元素进行分组。这在数据处理中是非常常见的一种情况。本文将会介绍如何在 Python 中实现按照符号对连续元素进行分组的功能。

问题

假设我们有一个包含连续数字的列表：

my_list = [1,2,3,6,7,9,10,11,15,16,17]

我们希望将这个列表按照连续的数字部分分组，得到的结果如下：

[[1, 2, 3], [6, 7], [9, 10, 11], [15, 16, 17]]

请问如何在 Python 中实现这个功能呢？

解决方案

一种解决方案是使用循环迭代整个列表，判断每个元素与前一个元素的差是否为 1，如果是的话，将其添加到当前连续序列中。否则将当前连续序列作为一个列表添加到结果列表中，重新开始新的连续序列。

def group_continuous_numbers(my_list):
    res = []
    continuous = []
    for num in my_list:
        if len(continuous) == 0 or num - continuous[-1] == 1:
            continuous.append(num)
        else:
            res.append(continuous)
            continuous = [num]
    if continuous:
        res.append(continuous)
    return res

现在我们可以使用上述函数来对 my_list 进行分组：

my_list = [1,2,3,6,7,9,10,11,15,16,17]
print(group_continuous_numbers(my_list))
# 输出：[[1, 2, 3], [6, 7], [9, 10, 11], [15, 16, 17]]

总结

本文介绍了如何在 Python 中实现按照符号对连续元素进行分组的功能。我们可以采用一些简单的方法来解决这个问题，这对于处理数据时非常有帮助。