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