以下是一个示例代码,展示了如何按值分组,直到达到阈值。
def group_by_threshold(values, threshold):
groups = [[]] # 初始化一个空组
current_group_sum = 0
for value in values:
if current_group_sum + value <= threshold:
groups[-1].append(value) # 将值添加到当前组
current_group_sum += value # 更新当前组的总和
else:
groups.append([value]) # 创建一个新的组
current_group_sum = value # 更新当前组的总和
return groups
# 示例用法
values = [1, 2, 3, 4, 5, 6, 7, 8, 9]
threshold = 10
result = group_by_threshold(values, threshold)
print(result)
在这个示例中,我们首先定义了一个group_by_threshold
函数,它接受两个参数:values
表示要分组的值的列表,threshold
表示每个组的最大总和阈值。
函数首先创建了一个空组groups
,并初始化了一个变量current_group_sum
来跟踪当前组的总和。然后,它遍历values
列表中的每个值。
对于每个值,如果将其添加到当前组后的总和仍然小于等于阈值,则将其添加到当前组,并更新当前组的总和。如果将其添加到当前组后的总和超过阈值,则创建一个新的组,并将该值添加到新组中。然后,将总和重置为该值。
最后,函数返回分组结果。
在示例用法中,我们定义了一个values
列表和一个threshold
值,并调用group_by_threshold
函数进行分组。最后,我们打印结果以验证分组是否正确。在这个例子中,输出应该是[[1, 2, 3, 4], [5, 6, 7], [8], [9]]
。