枚举类在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
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', , 1)
('Tue', , 2)
('Wed', , 3)
('Thu', , 4)
('Fri', , 5)
('Sat', , 6)
('Sun', , 7)
我们也可以通过下面方式访问对应日期的值,
sunday = Weekday.Sun.value
到这里,枚举类的使用就讲解完了。
Python是一门相对简单的编程语言,这里所谓的简单是入门简单,真正想成为一名高阶的Python开发者,有很多知识是书籍上都没有的。
就如同本文讲解的枚举类,或许你定义常量、使用字典都可以解决这个问题。但是,当遇到枚举问题时,再使用这些入门级的方法,对于代码的效率、可读性都会大打折扣。因此,当遇到一个需求,不应该把实现功能当成最终目标,而是应该把实现高性能系统当成最终追求。
我记得看过雷军的一个采访,他曾提到写的代码像诗一样,虽然网络上对他这句话的调侃很多。但是,我认为这是作为程序员应该有的追求,也是绝大多数程序员做不到的一点。
以上就是本次分享的所有内容,如果你觉得文章还不错,欢迎关注公众号:Python编程学习圈,每日干货分享,发送“J”还可领取大量学习资料,内容覆盖Python电子书、教程、数据库编程、Django,爬虫,云计算等等。或是前往编程学习网,了解更多编程技术知识。