Python中Enum使用的幾點註意事項

Posted on by

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使用註意事項內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：