如何从列表中获取每个项目的流入/流出？

Question

我有这个清单：

myList = [a, a, a, b, b, a, c, c, b, a]

我要计算每一项的流入量和流出量。你知道吗

‘a’的流入量=出现次数=5 （当过渡进入“a”时）

“a”的流出量=在“a”之后的不同字符数=2 （当从“a”到另一个字符的转换退出时）

对于我来说，我有这个，它是有效的：

myListDict = {}    
for item in myList:
    myListDict.setdefault(item, 0)
    myListDict[item] += 1

但我真的不知道如何在一个完整的迭代中，以一种快速而优雅的方式完成这项工作，如果可能的话。你知道吗

Answer 1

使用collections.Counter和itertools.groupby：

from collections import Counter
from itertools import groupby

def in_out_flow(lst):
    in_flow = Counter(lst)
    out_flow = Counter(k for k, _ in groupby(lst))
    out_flow[lst[-1]] -= 1
    for k,v in in_flow.items():
      print('key: {}, in flow: {}, out flow: {}'.format(k, v, out_flow[k]))

示例：

in_out_flow(['a', 'a', 'a', 'b', 'b', 'a', 'c', 'c', 'b', 'a'])
print('##')
in_out_flow(['a', 'a', 'a', 'b', 'a', 'c', 'a', 'b'])

输出：

key: a, in flow: 5, out flow: 2
key: b, in flow: 3, out flow: 2
key: c, in flow: 2, out flow: 1
##
key: a, in flow: 5, out flow: 3
key: b, in flow: 2, out flow: 1
key: c, in flow: 1, out flow: 1