python – 允许重复的计数器

发布时间：2020-12-16 22:31:20 所属栏目：Python 来源：网络整理

导读：我有一个交通灯枚举定义可能的状态： class TrafficLightPhase(Enum): RED = "RED" YELLOW = "YELLOW" GREEN = "GREEN" 我轮询一个交通信号灯每秒获取当前状态,我用这个函数将值放在一个双端队列中： def read_phases(): while running: current_phase = get

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

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

（编辑：李大同）

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