python - 如何使用 map 或 reduce 函数在 Python 中压缩列表？

Question

我想按照以下规则在 Python 中压缩一个列表:

['a', 'a', 'a', 'b', 'b', 'c']  ->  [3, 2, 1]

我想使用Python内置的map/reduce函数，如何实现？

谢谢!

Answer 1

使用itertools.groupby :

>>> import itertools
>>> [len(list(grp)) for key, grp in itertools.groupby(['a', 'a', 'a', 'b', 'b', 'c'])]
[3, 2, 1]
>>> [sum(1 for _ in grp) for key, grp in itertools.groupby(['a', 'a', 'a', 'b', 'b', 'c'])]
[3, 2, 1]

使用map、reduce:

>>> import operator
>>>
>>> def f(result, item):
...     if result and result[-1][0] == item:
...         return result[:-1] + [[item, result[-1][1]+1]]
...     else:
...         return result + [[item, 1]]
...
>>> map(operator.itemgetter(1), reduce(f, ['a', 'a', 'a', 'b', 'b', 'c'], []))
[3, 2, 1]