时间戳

首页 > 解决方案 > 在 Python3 中使用枚举作为字典键时出现“键错误”

python - 在 Python3 中使用枚举作为字典键时出现“键错误”

问题描述

我想使用枚举作为字典的键，但得到一个 KeyError。

#!/usr/bin/python3

from enum import Enum, unique
from typing import List

@unique
class Color(Enum):
    RED = "cherry"
    GREEN = "cucumber"
    BLUE = "blueberry"

allColors = {}

def countColors(colors: List[Color]):
    for c in colors:
        allColors[c] += 1

countColors([Color.RED, Color.RED, Color.BLUE, Color.GREEN])
for c in allColors:
    print(f"""{allColors[c]} {c.value} {c.name} pipes""")

当我运行这个时，我得到

Traceback (most recent call last):
  File "mvce.py", line 18, in <module>
    countColors([Color.RED, Color.RED, Color.BLUE, Color.GREEN])
  File "mvce.py", line 16, in countColors
    allColors[c] += 1
KeyError: <Color.RED: 'cherry'>

字典上的文档说我可以使用任何不可变值作为键，并且我假设枚举值是不可变的。

如何使用枚举作为字典中的键？

标签： pythonpython-3.xdictionaryenums

解决方案

会不会是你设置了 3 种颜色，但在第 18 行你有 4

countColors([Color.RED, Color.RED, Color.BLUE, Color.GREEN])

尝试

countColors ([Color.RED, Color.Blue, Color.Green])

推荐阅读