python中枚举有多种实现方式,如元组、全局变量(类似于c语言的宏)、枚举类等。个人感觉枚举类最易用,也最规范。
枚举类的定义与使用说明如下:
(1)从enum中导入Enum类,以继承Enum类;
(2)成员名称不允许重复,但值可以重复,相当于别名;值重复时,通过值只能取到第一个成员;
(3)若要限制值不可重复,需增加装饰器@unique;
(4)成员可进行同一性比较,但不可进行大小比较。
枚举的操作与字典类似,对于整个枚举结构,使用keys、values;但对于单个枚举值,对应name、value,而不是key、value。
如下的测试代码中,可清晰的看出枚举的基本用法。
测试代码:
@unique
class TLV_TAG(Enum):
STR = 1
INT = 2
# 取所有枚举成员的keys、values
print('keys : ', TLV_TAG.__members__.keys())
print('values : ', TLV_TAG.__members__.values())
print('items : ', TLV_TAG.__members__.items())
# 取枚举成员数量
print('length : ', len(TLV_TAG)) # 有警告,但可以用
print(TLV_TAG.__len__())
# 取值方式
print(TLV_TAG.INT)
print(TLV_TAG(2))
print(TLV_TAG['INT'])
# 判断一个枚举值是否属于该枚举
print(TLV_TAG.STR in TLV_TAG.__members__.values())
# 判断一个字符串是否属于该枚举的keys,有时需要将枚举与字符串进行转换
print('INT' in TLV_TAG.__members__.keys())
# 输出一个枚举值的name、value
print(TLV_TAG.INT.name, ' <----> ', TLV_TAG.INT.value)
# 遍历所有枚举值
for name, value in TLV_TAG.__members__.items():
print(name, ' ==> ', value)
输出
keys : odict_keys(['STR', 'INT'])
values : odict_values([<TLV_TAG.STR: 1>, <TLV_TAG.INT: 2>])
items : odict_items([('STR', <TLV_TAG.STR: 1>), ('INT', <TLV_TAG.INT: 2>)])
length : 2
2
TLV_TAG.INT
TLV_TAG.INT
TLV_TAG.INT
True
True
INT <----> 2
STR ==> TLV_TAG.STR
INT ==> TLV_TAG.INT