python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python Enum使用注意事项

Python中Enum使用的几点注意事项

作者:songofhawk

Python中的枚举是作为一个类存在的,这是与其他语言的一个较为鲜明的特征,下面这篇文章主要给大家介绍了关于Python中Enum使用的几点注意事项,需要的朋友可以参考下

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 对枚举类进行装饰

总结

到此这篇关于Python中Enum使用的几点注意事项的文章就介绍到这了,更多相关Python Enum使用注意事项内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文