python一堆数字相近的分成一组

海涛高软

于 2024-07-08 16:06:58 发布

阅读量159

点赞数 2

文章标签： python 开发语言

本文链接：https://blog.csdn.net/qq_15267341/article/details/140271248

版权

def group_nearest_numbers(numbers, threshold):
    groups = {}
    for num in numbers:
        found = False
        for group_id, group_members in groups.items():
            if abs(num - group_members[-1]) <= threshold:
                groups[group_id].append(num)
                found = True
                break
        if not found:
            groups[len(groups)] = [num]
    return groups.values()


# 示例使用
numbers = [1, 2, 3, 4, 500, 501, 502, 1000, 1001, 1002 ,1500,1600]
threshold = 100
groups = group_nearest_numbers(numbers, threshold)
for group in groups:
    print(group)

这个函数会将数字列表numbers中相差不超过threshold的数字分到同一组。最后，函数返回每个分组的列表。