我有一个交通灯枚举定义可能的状态：
@H_403_6@class TrafficLightPhase(Enum): RED = "RED" YELLOW = "YELLOW" GREEN = "GREEN"
我轮询一个交通信号灯每秒获取当前状态,我用这个函数将值放在一个双端队列中：
@H_403_6@def read_phases(): while running: current_phase = get_current_phase_phases() last_phases.append(current_phase) time.sleep(1)
我想对相同状态的序列进行分组,以便了解交通信号灯的相位时序.
我尝试使用Counter类的集合,如下所示：
@H_403_6@counter = collections.Counter(last_phases)
它组合了很好的不同状态,但我不知道下一个周期何时开始.是否有类似Counter的数据结构允许重复？所以我可以得到如下结果：
@H_403_6@Counter({ 'RED': 10,'GREEN': 10,'YELLOW': 3,'RED': 10,'YELLOW': 3 })
代替：
计数器({
‘RED’：30,
‘绿色’：30,
‘黄色’：9
})
最佳答案
我会使用itertools.groupby.它将对同一元素的连续运行进行分组,然后您可以检查每次运行的长度.
@H_403_6@>>> from itertools import groupby >>> last_phases= ['red','red','yellow','green'] >>> [(key,len(list(group))) for key,group in groupby(last_phases)] [('red',2),('yellow',1),('red',('green',1)]