Python Enum使用注意事项

软件发布|下载排行|最新软件

当前位置:首页IT学院IT技术

Python Enum使用注意事项

songofhawk   2022-05-23 我要评论

Enum 是个类

所以

基本的类操作都可以用

也就是我们可以添加自己的方法

class Mood(Enum):
    FUNKY = 1
    HAPPY = 3
    def describe(self):
        # self is the member here
        return self.name, self.value
    def __str__(self):
        return 'my custom str! {0}'.format(self.value)
    @classmethod
    def favorite_mood(cls):
        # cls here is the enumeration
        return cls.HAPPY
>>> Mood.favorite_mood()
<Mood.HAPPY: 3>
>>> Mood.HAPPY.describe()
('HAPPY', 3)
>>> str(Mood.FUNKY)
'my custom str! 1'

Enum的每个类成员,都会自动被转换成当前类的一个实例

from enum import Enum
class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3
>>> type(Color.RED)
<enum 'Color'>
>>> isinstance(Color.GREEN, Color)
True

这就意味着,我们不能用枚举成员直接当成它的value来用:

>>> Color.RED == 1
False
>>> Color.RED.value == 1
False

枚举成员还有个name属性,跟它的变量名相同

>>> Color.RED.name == "RED"
True

枚举类有个大坑:父类有成员的时候,就不能定义子类

所以,对于上面的Color类,如果还想定义一个子类,就会出错:

>>> class MoreColor(Color):
...     PINK = 17
...
Traceback (most recent call last):
...
TypeError: MoreColor: cannot extend enumeration 'Color'

但父类没有枚举成员,仅仅定义了函数是可以的:

class Foo(Enum):
    def some_behavior(self):
        pass

class Bar(Foo):
    HAPPY = 1
    SAD = 2

这显然大大限制了枚举类的扩展,一个不能访问成员的函数,用处也就不大了。

附Python Enum使用总结

1.枚举类不能用来实例化对象

2.访问枚举类中的某一项,直接使用类名访问加上要访问的项即可,比如 color.RED

3.枚举类里面定义的 Key = Value,在类外部不能修改 Value 值

4.枚举项可以用来比较,使用 ==,或者 is

5.枚举类中的 Key 不能相同,Value 可以相同,但是 Value 相同的各项 Key 都会当做别名

6.枚举类可以用 for 进行遍历,members.items() 可以遍历出含有别名的类

7.如果要枚举类中的 key 也不能相同,需要导入 unique 对枚举类进行装饰

总结

Copyright 2022 版权所有 软件发布 访问手机版

声明:所有软件和文章来自软件开发商或者作者 如有异议 请与本站联系 联系我们