python – 允许重复的计数器

我有一个交通灯枚举定义可能的状态：

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)]

（编辑：安卓应用网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!