在Python中应用枚举类，只需要这1个理由就够了！_python 枚举快还是字典快-CSDN博客

本文链接：https://blog.csdn.net/jakpopc/article/details/118372216

枚举类在Java中并不陌生，本文就来讲解一下为什么要在Python中应用枚举类。

试想一个问题，假如你在项目中处理通过Rest API请求回来的JSON数据，其中有一个数据格式如下：

{
  metrics: 'speed',
  value: 20,
  weekday: 'Mon'
}

你需要处理数据，把字符串型的周几转化成数值类型的1、2、3、4、5、6、7，你该怎么做？

Mon = 1
Tue = 2
Wed = 3
Thu = 4
Fri = 5
Sat = 6
Sun = 7

以定义常量的方式去预先定义一个一一对应的映射关系。

如果稍微有一定的Python开发经验的，大多数会选择字典的形式进行解决：

weekday = {
  'Mon': 1
  'Tue': 2
  'Wed': 3
  'Thu': 4
  'Fri': 5
  'Sat': 6
  'Sun': 7
}

通过键值的方式去取对应的数值型值。

但是，我在另外一篇文章中（从青铜到王者，一文教你节省90%内存占用）介绍过，Python中字典是一种非常占用内存的数据存储方式。

如果读过我的那篇文章，可能会产生很多新的想法，例如，用类的方式去实现枚举关系，

class Weekday:
  Mon = 1
  Tue = 2
  Wed = 3
  Thu = 4
  Fri = 5
  Sat = 6
  Sun = 7

但是，上述这些方式在数据持久化和扩展性方面并不是最理想的解决方案。

如果有一定的Java开发经验，面对这样的问题，肯定会马上想起一个概念--枚举类。

enum Weekday {
    MON(1, "星期一"),
    TUE(2, "星期二"),
    WED(3, "星期三"),
    THU(4, "星期四"),
    FRI(5, "星期五"),
    SAT(6, "星期六"),
    SUN(7, "星期七");

    publicfinalint intValue;
    publicfinal String chinese;

    Weekday(int intValue, String chinese) {
        this.intValue = intValue;
        this.chinese = chinese;
    }
}

通过下面方式，就可以访问枚举类中的值，

publicclass App {
    public static void main(String[] args) {
        Weekday day = Weekday.MON;
        System.out.println(day.intValue);
        System.out.println(day.chinese);
    }
}

Java中枚举类具有众多优势，例如，安全、规范，使得它备受欢迎。

那么，Python中是否也可以使用枚举类解决类似的枚举问题？

答案是肯定的。

Python中提供了标准的enum库，通过对它的调用和继承，就可以实现Python枚举类。

实现方式有2种。

方式1：类继承

# demo.py

from enum import Enum

class Weekday(Enum):
  Mon = 1
  Tue = 2
  Wed = 3
  Thu = 4
  Fri = 5
  Sat = 6
  Sun = 7
  
print(Weekday)
print(Weekday.Mon)
print(Weekday.Mon.value)

输出结果看一下，

$ python demo.py
<enum 'Weekday'>
Weekday.Mon
1

方式2：定义变量

当每个枚举的值是递增、且不重要时，可以使用常量名作为字符串直接调用Enum类，

Weekday = Weekday('Weekday', ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun'])
for name, member in Weekday.__members__.items():
	print(name, member, member.value)

下面，输出结果看一下，

$ python demo.py
('Mon', <Weekday.Mon: 1>, 1)
('Tue', <Weekday.Tue: 2>, 2)
('Wed', <Weekday.Wed: 3>, 3)
('Thu', <Weekday.Thu: 4>, 4)
('Fri', <Weekday.Fri: 5>, 5)
('Sat', <Weekday.Sat: 6>, 6)
('Sun', <Weekday.Sun: 7>, 7)

我们也可以通过下面方式访问对应日期的值，