python学习:枚举定义和使用

python学习:枚举定义和使用

1.定义

在某些情况下,一个类的对象是有限且固定的,比如季节类,它只有 4 个对象;再比如行星类,目前只有 8 个对象。这种实例有限且固定的类,在 Python 中被称为枚举类。
程序有两种方式来定义枚举类:

  • 直接使用 Enum 列出多个枚举值来创建枚举类。
  • 通过继承 Enum 基类来派生枚举类。

2.使用

使用 Enum 列出多个枚举值来创建枚举类

如下程序示范了直接使用 Enum 列出多个枚举值来创建枚举类:

import enum
#定义枚举类
Season = enum.Enum("Season", ("SPRING", "SUMMER", "FALL", "WINTER"))

上面程序使用 Enum() 函数(就是 Enum 的构造方法)来创建枚举类,该构造方法的第一个参数是枚举类的类名;第二个参数是一个元组,用于列出所有枚举值。

在定义了上面的 Season 枚举类之后,程序可直接通过枚举值进行前问,这些枚举值都是该枚举的成员,每个成员都有 name、value 两个属性,其中 name 属性值为该枚举值的变量名,value 代表该枚举值的序号(序号通常从 1 开始)。

例如,如下代码测试了枚举成员的用法:

#直接访问指定枚举对象
print(Season.SPRING)
 
#访问枚举成员的变量名
print(Season.SPRING.name)
 
#访问枚举成员的值
print(Season.SPRING.value)

输出如下:

Season.SPRING
SPRING
1

通过继承 Enum 基类来派生枚举类

也可通过枚举变量名或枚举值来访问指定枚举对象。

例如如下代码:

#根据枚举变量名访问枚举对象
print(Season["SUMMER"])    #输出Season.SUMMER
  
#根据枚举值访问枚举对象       #输出Season.FALL
print(Season(3))

  

Python 还为枚举提供了一个 __members__ 属性,该属性返回一个 dict 字典,字典包含了该枚举的所有枚举实例。
程序可通过遍历 __members__ 属性来访问枚举的所有实例。

例如如下代码:

#遍历所有的成员
for name, member in Season.__members__.items():
    print(member, "--->", name,"--->", member.value)
 
 
#输出如下:
Season.SPRING ---> SPRING ---> 1
Season.SUMMER ---> SUMMER ---> 2
Season.FALL ---> FALL ---> 3
Season.WINTER ---> WINTER ---> 4

如果要定义更复杂的枚举,则可通过继承 Enum 来派生枚举类,在这种方式下程序就可以为枚举额外定义方法了。

例如如下程序:

import enum
class Orientation(enum.Enum):
    #为序列指定值
    EAST = "东"
    SOUTH = "南"
    WEST = "西"
    NORTH = "北"
 
    def info(self):
        print("这是一个定义方向为【%s】的枚举"% self.value)
 
print(Orientation.SOUTH)
print(Orientation.SOUTH.value)
#通过枚举变量访问枚举
print(Orientation["WEST"])
 
#通过枚举值访问枚举
print(Orientation("北"))
 
#调用枚举的info()方法
Orientation.EAST.info()
 
#循环遍历Orientation枚举的所有成员
for name, member in Orientation.__members__.items():
    print(member, "--->", name, "--->", member.value)
"""
学习中遇到问题没人解答?小编创建了一个Python学习交流群:711312441
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
""" 
 
 
 
#输出如下:
Orientation.SOUTH
南
Orientation.WEST
Orientation.NORTH
这是一个定义方向【东】的枚举
Orientation.EAST ---> EAST ---> 东
Orientation.SOUTH ---> SOUTH ---> 南
Orientation.WEST ---> WEST ---> 西
Orientation.NORTH ---> NORTH ---> 北

上面程序通过继承 Enum 派生了 Orientation 枚举类,通过这种方式派生的枚举类既可额外定义方法,如上面的 info() 方法所示,也可为枚举指定 value(value 的值默认是 1、2、3、…)。

虽然此时 Orientation 枚举的 value 是由类型,但该枚举同样可通过 value 来访问特定枚举,如上面程序中的 Orientation(“南”),这是完全允许的。

3.枚举构造器

枚举也是类,因此枚举也可以定义构造器。为枚举定义构造器之后,在定义枚举实例时必须为构造器参数设置值。

例如如下程序:import enum

class Gender(enum.Enum):
    MALE = "男", "阳刚之力"
    FEMALE = "女", "柔顺之美"
 
    def __init__(self, cn_name, desc):
        self._cn_name = cn_name
        self._desc = desc
 
    @property
    def desc(self):
        return self._desc
 
    @property
    def cn_name(self):
        return self._cn_name
 
#访问FEMALE的name
print("FEMALE的name:", Gender.FEMALE.name)
 
#访问FEMALE的value
print("FEMALE的value:", Gender.FEMALE.value)
 
#访问自定义的cn_name属性
print("FEMALE的cn_name:", Gender.FEMALE.cn_name)
 
# 访问自定义的desc属性
print("FEMALE的desc:", Gender.FEMALE.desc)<br><br>

上面程序定义了 Gender 枚举类,并为它定义了一个构造器,调用该构造器需要传入 cn_name 和 desc 两个参数,因此程序使用如下代码来定义 Gender 的枚举值。

MALE = "男", "阳刚之力"
FEMALE = "女", "柔顺之美

上面代码为 MALE 枚举指定的 value 是‘男’和‘阳刚之力’这两个字符串,其实它们会被自动封装成元组后传给 MALE 的 value 属性;而且此处传入的‘男’和‘阳刚之力’ 这两个参数值正好分别传给 cnname 和 desc 两个参数。

简单来说,枚举的构造器需要几个参数,此处就必须指定几个值。

输出如下:

FEMALE的name: FEMALE
FEMALE的value: ("女", "柔顺之美")
FEMALE的cn_name: 女
FEMALE的desc: 柔顺之美
hmoban主题是根据ripro二开的主题,极致后台体验,无插件,集成会员系统
自学咖网 » python学习:枚举定义和使用