Python在子列表中分组列表中的重复值

Question

我需要从一个列表中追加一些重复的值到一个子列表中，让我用一个例子来解释：

我有一个名为array的变量，它包含大写字母字符串和$符号。

array = ['F', '$', '$', '$', 'D', '$', 'C']

我的最终目标是拥有这个数组：

final_array = ['F', ['$', '$', '$'], 'D', ['$'], 'C']

在示例中，我需要将所有的$符号分组到原始数组中的子列表中，我想过在数组中迭代，找到当前$附近的所有符号，然后创建第二个数组，但我想也许还有更多的pythonic我可以做的，有什么想法吗？

Answer 1

您可以从itertools中使用groupby

array = ['F', '$', '$', '$', 'D', '$', 'C']

from itertools import groupby

result = []
for key, group in groupby(array):
    if key == '$':
        result.append(list(group))
    else:
        result.append(key)

print(result)

当然，您可以将for循环缩短为一个理解：

result = [list(group) if key == '$' else key for key, group in groupby(array)]