溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這期內容當中小編將會給大家帶來有關深入淺析Python中的枚舉，文章內容豐富且以專業的角度為大家分析和敘述，閱讀完這篇文章希望大家可以有所收獲。
enum 是一組綁定到唯一常數值的符號名稱，并且具備可迭代性和可比較性的特性。我們可以使用 enum 創建具有良好定義的標識符，而不是直接使用魔法字符串或整數，也便于開發工程師的代碼維護。
創建枚舉
我們可以使用 class 語法創建一個枚舉類型，方便我們進行讀寫，另外，根據函數 API 的描述定義，我們可以創建一個 enum 的子類，如下：
from enum import Enum class HttpStatus(Enum): OK = 200 BAD_REQUEST = 400 FORBIDDEN = 403 NOT_FOUND = 404 REQUEST_TIMEOUT = 408 SERVICE_UNAVAILABLE = 500
注意： 枚舉屬性值可以是任何東西: int, str 等。如果確切的值不重要，您可以使用 auto 實例，并為您選擇適當的值。如果您將 auto 與其他值混合，則必須小心。 枚舉類型中，不可以設置相同名稱的 name，可以有相同的 value。
enum 自帶屬性 name 和 value，日常工作中使用最多的也是這兩個屬性，我們打印看看結果：
print('Member: {}'.format(HttpStatus.OK)) # Member: HttpStatus.OK
print('Member name: {}'.format(HttpStatus.OK.name)) # Member name: OK
print('Member value: {}'.format(HttpStatus.OK.value)) # Member value: 200
print(repr(HttpStatus.OK)) # <enum 'HttpStatus'>
print(type(HttpStatus.OK)) # <HttpStatus.OK: 200>
print(isinstance(HttpStatus.OK, HttpStatus)) # True枚舉迭代
枚舉支持迭代和遍歷順序。舉個例子：
from enum import Enum, auto
# 創建
class HttpStatus(Enum):
OK = 200
BAD_REQUEST = 400
FORBIDDEN = 403
NOT_FOUND = 404
REQUEST_TIMEOUT = 408
SERVICE_UNAVAILABLE = 500
OTHER = auto.value
# 迭代
for status in HttpStatus:
print('{} : {}'.format(status.name, status.value))打印結果：
OK : 200
BAD_REQUEST : 400
FORBIDDEN : 403
NOT_FOUND : 404
REQUEST_TIMEOUT : 408
SERVICE_UNAVAILABLE : 500
OTHER : <object object at 0x000002863E1D7B10>
可以看出，遍歷的每一個 status 是一個獨立的枚舉成員，擁有 name 和 value 屬性。
另外，我們也可以使用如下形式來進行枚舉遍歷：
for name, member in HttpStatus.__members__.items():
print('{} : {}'.format(name, member))枚舉成員與屬性訪問
通過枚舉 value 進行訪問，訪問需要使用元組()的形式
print(HttpStatus(200)) # HttpStatus.OK
通過枚舉 name 進行訪問，訪問需要使用列表[]的形式
print(HttpStatus['OK']) # HttpStatus.OK
將屬性賦予另一個 enum 成員
number = HttpStatus.OK print(number) # HttpStatus.OK
枚舉值唯一
上面我們創建的枚舉類中，value 值是可以重復的，如果我們不想枚舉類中的值重復可以是用裝飾器 @unique，舉例如下：
from enum import Enum, unique # 創建 @unique class HttpStatus(Enum): OK = 200 BAD_REQUEST = 400 FORBIDDEN = 403 NOT_FOUND = 404 REQUEST_TIMEOUT = 408 SERVICE_UNAVAILABLE = 500 OTHER = 200
我們運行后，報如下異常：
ValueError: duplicate values found in <enum 'HttpStatus'>: OTHER -> OK
我們查看源代碼，發現加入此裝飾器的枚舉類型，unique 方法會將其 __members__.items() 進行遍歷，追加到 duplicates 列表中，如果發現列表不為空，則拋出如上異常信息。
枚舉自動賦值
此功能用于我們在使用枚舉時，只在意枚舉的標識符的含義而不在意值的情況下，但是如果需要與字符串或整數混合使用就要額外注意。下面貼上官方的示例：
import unittest
from enum import auto, Enum
class TestEnum(unittest.TestCase):
def test_auto_number(self):
class Color(Enum):
red = auto()
blue = auto()
green = auto()
self.assertEqual(list(Color), [Color.red, Color.blue, Color.green])
self.assertEqual(Color.red.value, 1)
self.assertEqual(Color.blue.value, 2)
self.assertEqual(Color.green.value, 3)
def test_auto_name(self):
class Color(Enum):
def _generate_next_value_(self, start, count, last):
return self
red = auto()
blue = auto()
green = auto()
self.assertEqual(list(Color), [Color.red, Color.blue, Color.green])
self.assertEqual(Color.red.value, 'red')
self.assertEqual(Color.blue.value, 'blue')
self.assertEqual(Color.green.value, 'green')可以發現，使用 auto() 得到的是整數自增型，如果我們需要別的方式，只需要在我們的枚舉類中，重寫 _generate_next_value_ 方法。
枚舉比較
枚舉對象可以進行比較，但是不能進行值比較，如果需要進行值比較則需要枚舉對象繼承 IntEnum 對象，舉個例子：
import unittest
from enum import Enum, IntEnum
class TestEnum(unittest.TestCase):
class Season(IntEnum):
SPRING = 1
SUMMER = 2
AUTUMN = 3
WINTER = 4
def test_comparisons(self):
season = self.Season
self.assertEqual(season.SPRING, 1)
class Part(Enum):
SPRING = 1
CLIP = 2
BARREL = 3
self.assertNotEqual(Part.SPRING, 1)
self.assertNotEqual(Part.SPRING, season.SPRING)
TestEnum().test_comparisons()上面的測試例子當中，我們創建了兩個繼承類型不一樣的枚舉類，可以看到繼承了 IntEnum 的 Season 可以進行值的比較，而繼承了 Enum 的 Part 則不能進行值比較，并且 IntEnum 類型與 Enum 類型也不能進行比較，即使屬性和值一樣。
枚舉方法
枚舉中可以定義枚舉類自身特有的方法，也可以復寫一些已經在基類中定義好的方法，比如: __init__, __str__, __repr__ , __hash__ ,__format__ 等。舉個例子：
from enum import Enum
class Mood(Enum):
FUNKY = (1, "hello")
HAPPY = (3, "world")
def describe(self):
return self.name, self.value
def __init__(self, num, nice):
self.num = num
self.nice = nice
def __str__(self):
return 'my custom str! {0}'.format(self.value)
@classmethod
def favorite_mood(cls):
return cls.HAPPY
@property
def testValue(self):
return self.nice + ':' + str(self.num)上面我們定義了一個枚舉類，其中 value 是一個枚舉類型，我們可以定義 __init__ 方法去對應元組中的值，我們也復寫了 __str__ 方法。
打印方法看看效果：
print(Mood.favorite_mood()) # my custom str! (3, 'world')
print(Mood.HAPPY.describe()) # ('HAPPY', (3, 'world'))
print(str(Mood.FUNKY)) # my custom str! (1, 'hello')
print(Mood.FUNKY.testValue) # hello:1從輸出結果看，我們自定義和復寫的方法都已經成功的應用到了 Mood 類中。
枚舉繼承
不同于 java 中的枚舉類, python 中的枚舉類是可以被繼承的，但是被繼承的枚舉類規定其不能定義任何成員，但可以定義抽象方法。舉例如下：
class EnumExtend(unittest.TestCase):
def test_extending(self):
class Shade(Enum):
def shade(self):
print(self.name)
class Color(Shade):
red = 1
green = 2
blue = 3
with self.assertRaises(TypeError):
class MoreColor(Color):
cyan = 4
magenta = 5
yellow = 6
def test_extending2(self):
class Shade(Enum):
def shade(self):
return self.name
class Color(Shade):
def hex(self):
return '%s nice!' % self.value
class MoreColor(Color):
cyan = 4
magenta = 5
yellow = 6
self.assertEqual(MoreColor.magenta.shade(), 'magenta')
self.assertEqual(MoreColor.magenta.hex(), '5 nice!')測試用例可以完美運行，我們可以發現：第一個方法中，拋出了 TypeError 的異常；第二個測試方法中，MoreColor 繼承了 Color， Color 繼承了 Shade, 并且我們可以通過子類調用父類中的方法。
上述就是小編為大家分享的深入淺析Python中的枚舉了，如果剛好有類似的疑惑，不妨參照上述分析進行理解。如果想知道更多相關知識，歡迎關注億速云行業資訊頻道。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
