python - 合并 Python 列表中的条目

Question

假设我有一个列表 test = ['a','b','b','b','a','b','a','a','b', 'b']。如果我想将所有相邻的 'b' 组合在一起得到 test2 = ['a','b','a','b','a','a' ,'b']，那么最有效和优雅的做法是什么？

Answer 1

使用 itertools.groupby，效果如下:

>>> test = ['a','b','b','b','a','b','a','a','b','b']
>>> import itertools
>>> grouped = itertools.groupby(test)
>>> test2 = []
>>> for k, g in grouped:
...     if k == 'b':
...         test2.append(k)
...     else:
...         test2.extend(g)
...
>>> test2
['a', 'b', 'a', 'b', 'a', 'a', 'b']