python枚举类型定义与使用

末蓝、 2024-03-30 09:22 86阅读 0赞

在python中枚举是一种类（Enum,IntEnum），存放在enum模块中。枚举类型可以给一组标签赋予一组特定的值，本文给大家介绍的非常详细，对大家的学习或工作具有一定的帮助。

### 定义 ###

在某些情况下，一个类的对象是有限且固定的，比如季节类，它只有 4 个对象；再比如行星类，目前只有 8 个对象。这种实例有限且固定的类，在 Python 中被称为枚举类。

程序有两种方式来定义枚举类：

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

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

### 2.使用 ###

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

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

<table> 
 <tbody> 
  <tr> 
   <td> <p>1</p> <p>2</p> <p>3</p> </td> 
   <td> <p><code>import</code> <code>enum</code></p> <p><code>#定义枚举类</code></p> <p><code>Season </code><code>=</code> <code>enum.Enum(</code><code>'Season'</code><code>, (</code><code>'SPRING'</code><code>, </code><code>'SUMMER'</code><code>, </code><code>'FALL'</code><code>, </code><code>'WINTER'</code><code>))</code></p> </td> 
  </tr> 
 </tbody> 
</table>

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

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

例如，如下代码测试了枚举成员的用法：

<table> 
 <tbody> 
  <tr> 
   <td> <p>1</p> <p>2</p> <p>3</p> <p>4</p> <p>5</p> <p>6</p> </td> 
   <td> <p><code>#直接访问指定枚举对象</code></p> <p><code>print</code><code>(Season.SPRING)</code></p> <p><code>#访问枚举成员的变量名</code></p> <p><code>print</code><code>(Season.SPRING.name)</code></p> <p><code>#访问枚举成员的值</code></p> <p><code>print</code><code>(Season.SPRING.value)</code></p> </td> 
  </tr> 
 </tbody> 
</table>

输出如下：

> Season.SPRING  
> SPRING  
> 1

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

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

例如如下代码：

<table> 
 <tbody> 
  <tr> 
   <td> <p>1</p> <p>2</p> <p>3</p> <p>4</p> </td> 
   <td> <p><code>#根据枚举变量名访问枚举对象</code></p> <p><code>print</code><code>(Season[</code><code>'SUMMER'</code><code>])&nbsp;&nbsp;&nbsp; </code><code>#输出Season.SUMMER</code></p> <p><code>#根据枚举值访问枚举对象&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; #输出Season.FALL</code></p> <p><code>print</code><code>(Season(</code><code>3</code><code>))</code></p> </td> 
  </tr> 
 </tbody> 
</table>

Python 还为枚举提供了一个 `__members__`属性，该属性返回一个 dict 字典，字典包含了该枚举的所有枚举实例。

程序可通过遍历 `__members__`属性来访问枚举的所有实例。

例如如下代码：

<table> 
 <tbody> 
  <tr> 
   <td> <p>1</p> <p>2</p> <p>3</p> </td> 
   <td> <p><code>#遍历所有的成员</code></p> <p><code>for</code> <code>name, member </code><code>in</code> <code>Season.__members__.items():</code></p> <p><code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(member, </code><code>'---&gt;'</code><code>, name,</code><code>'---&gt;'</code><code>, member.value)</code></p> </td> 
  </tr> 
 </tbody> 
</table>

> \#输出如下：  
> Season.SPRING ---> SPRING ---> 1  
> Season.SUMMER ---> SUMMER ---> 2  
> Season.FALL ---> FALL ---> 3  
> Season.WINTER ---> WINTER ---> 4

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

转自：[https://www.weidianyuedu.com][https_www.weidianyuedu.com]

[https_www.weidianyuedu.com]: https://www.weidianyuedu.com