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的每個(gè)類成員,都會(huì)自動(dòng)被轉(zhuǎn)換成當(dāng)前類的一個(gè)實(shí)例

from enum import Enum
class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

>>> type(Color.RED)
<enum 'Color'>
>>> isinstance(Color.GREEN, Color)
True

這就意味著，我們不能用枚舉成員直接當(dāng)成它的value來用：

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

枚舉成員還有個(gè)name屬性，跟它的變量名相同

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

枚舉類有個(gè)大坑：父類有成員的時(shí)候，就不能定義子類

所以，對(duì)于上面的Color類，如果還想定義一個(gè)子類，就會(huì)出錯(cuò)：

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

但父類沒有枚舉成員，僅僅定義了函數(shù)是可以的：

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

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

這顯然大大限制了枚舉類的擴(kuò)展，一個(gè)不能訪問成員的函數(shù)，用處也就不大了。

附Python Enum使用總結(jié)

1.枚舉類不能用來實(shí)例化對(duì)象

2.訪問枚舉類中的某一項(xiàng)，直接使用類名訪問加上要訪問的項(xiàng)即可，比如 color.RED

3.枚舉類里面定義的 Key = Value，在類外部不能修改 Value 值

4.枚舉項(xiàng)可以用來比較，使用 ==，或者 is

5.枚舉類中的 Key 不能相同，Value 可以相同，但是 Value 相同的各項(xiàng) Key 都會(huì)當(dāng)做別名

6.枚舉類可以用 for 進(jìn)行遍歷，members.items() 可以遍歷出含有別名的類

7.如果要枚舉類中的 key 也不能相同，需要導(dǎo)入 unique 對(duì)枚舉類進(jìn)行裝飾

總結(jié)

到此這篇關(guān)于Python中Enum使用的幾點(diǎn)注意事項(xiàng)的文章就介紹到這了,更多相關(guān)Python Enum使用注意事項(xiàng)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: