1.枚舉的定義

首先，定義枚舉要導(dǎo)入enum模塊。

枚舉定義用class關(guān)鍵字，繼承Enum類。

用于定義枚舉的class和定義類的class是有區(qū)別【下一篇博文繼續(xù)分享】。

示例代碼:

fromenumimportEnum

classColor(Enum):

red=1

orange=2

yellow=3

green=4

blue=5

indigo=6

purple=7

代碼分析：

上面的代碼，我們定義了顏色的枚舉Color.

顏色枚舉有7個(gè)成員，分別是Color.red、Color.orange、Color.yellow等。

每一個(gè)成員都有它們各自名稱和值，Color.red成員的名稱是：red，值是：1。

每個(gè)成員的數(shù)據(jù)類型就是它所屬的枚舉。【*注：用class定義的類，實(shí)際上就是一種類型】

1.1定義枚舉時(shí)，成員名稱不允許重復(fù)

fromenumimportEnum

classColor(Enum):

red=1

red=2

上面的代碼，就無法執(zhí)行。提示錯(cuò)誤：TypeError:Attemptedtoreusekey:'red'

1.2默認(rèn)情況下，不同的成員值允許相同。但是兩個(gè)相同值的成員，第二個(gè)成員的名稱被視作第一個(gè)成員的別名

fromenumimportEnum

classColor(Enum):

red=1

red_alias=1

成員Color.red和Color.red_alias具有相同的值，那么成員Color.red_alias的名稱red_alias就被視作成員Color.red名稱red的別名。

1.3如果枚舉中存在相同值的成員，在通過值獲取枚舉成員時(shí)，只能獲取到第一個(gè)成員

fromenumimportEnum

classColor(Enum):

red=1

red_alias=1

print(Color(1))

輸出結(jié)果為：Color.red

1.4如果要限制定義枚舉時(shí)，不能定義相同值的成員。可以使用裝飾器@unique【要導(dǎo)入unique模塊】

fromenumimportEnum,unique

@unique

classColor(Enum):

red=1

red_alias=1

再執(zhí)行就會(huì)提示錯(cuò)誤：ValueError:duplicatevaluesfoundin:red_alias->red

2.枚舉取值

2.1通過成員的名稱來獲取成員

Color['red']

2.2通過成員值來獲取成員

Color(2)

2.3通過成員，來獲取它的名稱和值

red_member=Color.red

red_member.name

red_member.value

3.迭代器

3.1枚舉支持迭代器，可以遍歷枚舉成員

forcolorinColor:

print(color)

輸出結(jié)果是，枚舉的所有成員。Color.red、Color.orange、Color.yellow、Color.green、Color.blue、Color.indigo、Color.purple。

3.2如果枚舉有值重復(fù)的成員，循環(huán)遍歷枚舉時(shí)只獲取值重復(fù)成員的第一個(gè)成員

fromenumimportEnum

classColor(Enum):

red=1

orange=2

yellow=3

green=4

blue=5

indigo=6

purple=7

red_alias=1

forcolorinColor:

print(color)

輸出結(jié)果是：Color.red、Color.orange、Color.yellow、Color.green、Color.blue、Color.indigo、Color.purple。但是Color.red_alias并沒有出現(xiàn)在輸出結(jié)果中。

3.3如果想把值重復(fù)的成員也遍歷出來，要用枚舉的一個(gè)特殊屬性__members__

fromenumimportEnum

classColor(Enum):

red=1

orange=2

yellow=3

green=4

blue=5

indigo=6

purple=7

red_alias=1

forcolorinColor.__members__.items():

print(color)

輸出結(jié)果：('red',)、('orange',)、('yellow',)、('green',)、('blue',)、

('indigo',)、('purple',)、('red_alias',)

4.枚舉比較

4.1枚舉成員可進(jìn)行同一性比較

Color.redisColor.red

輸出結(jié)果是：True

Color.redisnotColor.blue

輸出結(jié)果是：True

4.2枚舉成員可進(jìn)等值比較

Color.blue==Color.red

輸出結(jié)果是：False

Color.blue!=Color.red

輸出結(jié)果是：True

4.3枚舉成員不能進(jìn)行大小比較

Color.red

輸出結(jié)果出錯(cuò)：TypeError:unorderabletypes:Color()

以上內(nèi)容為大家介紹了Python枚舉，希望對(duì)大家有所幫助，如果想要了解更多Python相關(guān)知識(shí)，請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。