关于python:在Python中应用枚举类只需要这1个理由就够了

40次阅读

共计 2121 个字符，预计需要花费 6 分钟才能阅读完成。

枚举类在 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)

咱们也能够通过上面形式拜访对应日期的值，