Python – 将每个连续的重复项加倍

有时我们需要将重复的元素在列表中处理一下，例如每个连续的重复项加倍。这可以通过Python中的一些内置功能来实现。

方法一：使用循环

def double_items(lst):
    result = []
    i = 0
    while i < len(lst):
        if i == len(lst) - 1 or lst[i] != lst[i+1]:
            result.append(lst[i])
        else:
            result.append(lst[i]*2)
            i += 1
        i += 1
    return result

lst = [1, 2, 2, 3, 4, 4, 4, 5, 5]
print(double_items(lst)) # 输出 [1, 4, 2, 3, 8, 4, 5, 10]

上述函数中，我们利用了while循环和if条件语句来逐个检查列表中的元素。如果当前元素和后一个元素相等，我们就将当前元素的值乘以2并追加到结果列表中；否则，我们就将当前元素追加到结果列表中。

方法二：使用itertools.groupby

import itertools

def double_items(lst):
    result = []
    for k, g in itertools.groupby(lst):
        group = list(g)
        result.extend(group if len(group) == 1 else [group[0]*2])
    return result

lst = [1, 2, 2, 3, 4, 4, 4, 5, 5]
print(double_items(lst)) # 输出 [1, 4, 2, 3, 8, 4, 5, 10]

上述函数中，我们使用了Python中的itertools模块中的groupby函数。该函数将相邻的重复元素放在一起，形成一个带有key和group的元组，并通过for循环依次读取。在for循环中，我们将当前组的第一个元素加入结果列表，并在后续组中的元素长度为1时，在结果列表中追加该元素；在后续组的元素长度大于1时，我们将该组的第一个元素乘以2并追加到结果列表中。

无论使用哪种方法，上述代码都可以将每个连续的重复项加倍。