Python使用enum模块获取枚举成员索引号的四种方法详解_Python

在 python 中，可以使用 enum 模块创建枚举类型，并通过遍历枚举成员来获取其索引号（即枚举值的序号）。以下是详细方法和示例：

方法 1：使用enum.enum和enumerate()

通过 enumerate() 遍历枚举成员，同时获取索引和枚举值。

from enum import enum

class color(enum):
    red = 1
    green = 2
    blue = 3

# 遍历枚举并获取索引和值
for index, color in enumerate(color, start=1):  # start=1 表示索引从1开始（默认从0）
    print(f"index: {index}, name: {color.name}, value: {color.value}")

输出：

index: 1, name: red, value: 1
index: 2, name: green, value: 2
index: 3, name: blue, value: 3

方法 2：直接通过枚举成员的name和value属性

如果不需要显式遍历索引，可以直接访问枚举成员的属性：

for color in color:
    print(f"name: {color.name}, value: {color.value}")

输出：

name: red, value: 1
name: green, value: 2
name: blue, value: 3

方法 3：获取枚举成员的序号（基于定义顺序）

如果需要枚举成员的定义顺序序号（从 0 开始），可以通过 list(enum).index(member) 实现：

for color in color:
    index = list(color).index(color)  # 获取定义顺序的索引（从0开始）
    print(f"order index: {index}, name: {color.name}, value: {color.value}")

输出：

order index: 0, name: red, value: 1
order index: 1, name: green, value: 2
order index: 2, name: blue, value: 3

方法 4：使用members属性

通过 __members__ 访问枚举的名称到成员的映射，再结合 enumerate()：

for i, name in enumerate(color.__members__, start=1):
    color = color[name]
    print(f"index: {i}, name: {name}, value: {color.value}")

输出：

index: 1, name: red, value: 1
index: 2, name: green, value: 2
index: 3, name: blue, value: 3

知识扩展

在 python 的 enum 模块中，枚举成员本身并不直接提供“索引”属性，但我们可以利用枚举类的内部顺序或成员的值来获得索引。常见的获取索引的方法有以下四种，具体使用哪种取决于你的枚举定义方式（是否有自定义值）。

1. 利用 list() 转换枚举类 + index() 方法

枚举类的 __members__ 保留成员定义顺序，可以通过将其转换为列表，再使用 index() 方法获取成员的位置（从 0 开始）。

from enum import enum
class color(enum):
    red = 1
    green = 2
    blue = 3
members = list(color)
idx = members.index(color.green)
print(idx)  # 输出 1

优点：不依赖成员的具体值，只要定义顺序不变即可。

注意：索引从 0 开始，若希望从 1 开始，加 1 即可。

2. 使用 _member_names_ 属性获取名称列表

枚举类内置了 _member_names_ 属性，它是一个按定义顺序排列的成员名称列表。可以通过成员名来定位索引。

names = color._member_names_
idx = names.index(color.green.name)
print(idx)  # 输出 1

说明：_member_names_ 是 python 3.11+ 公开的但并非“官方推荐”的 api，在旧版本中也可用（但内部属性）。它的用途明确，不过在实际项目中更推荐使用第一种方式。

3. 使用 __members__ 有序字典

__members__ 是一个保留定义顺序的映射（python 3.6+ 字典有序）。可以通过将其值转换为列表来获得索引。

members_list = list(color.__members__.values())
idx = members_list.index(color.blue)
print(idx)  # 输出 2

优点：直接利用标准属性，不依赖内部名称列表。

与第一种方法的区别：第一种本质也是 list(color)，它会自动调用 __members__.values() 的转换。

4. 借助 intenum 将索引作为成员值

如果希望索引等于成员的值，可以使用 enum.intenum（或 enum.enum 配合 auto()），让枚举值自动从 1 递增，那么成员的值本身就是索引（从 1 开始）。

from enum import intenum, auto
class color(intenum):
    red = auto()   # 1
    green = auto() # 2
    blue = auto()  # 3
print(color.green.value)   # 2（索引）

特点：此时成员的值严格等于定义顺序（索引）。但要注意 intenum 的所有成员可以直接与整数比较，可能会有副作用。如果不希望这种比较特性，可以只用 enum + auto() 但通过 value 获取索引。

对比总结