EasyDict可以让你像访问属性一样访问dict里的变量。

1. 问题

d = {'foo':3, 'bar':{'x':1, 'y':2}}
print(d['foo'])  # 如何想要访问字典的元素需要这么写
print(d['bar']['y'])  # 如果想要继续访问字典中字典的元素需要使用二维数组
# print(d.foo) 这样写会出错哦!

输出:

3
2

但是感觉这样太麻烦了,有没有更简单的方法使用字典了?

2. 解决方法

我们可以使用easydict模块!

from easydict import EasyDict as edict
easy = edict(d = {'foo':3, 'bar':{'x':1, 'y':2}}) # 将普通的字典传入到edict()
print(easy['foo'])  # 这是传统的方法
print(easy.foo)  # 这是我们使用easydict输出二者结果是一样的,但是可以更为方便的使用字典了
print(easy.bar.x)  # 我们也是可以很方便的使用字典中字典的元素了

输出:

3
3
1

3. Easydict的其他用法

3. 1.设置属性

from easydict import EasyDict as edict
d = edict()  # 这个是输出{}
d.foo = 3  # 我们可以直接赋值语句对字典元素进行创建
d.bar = {'prob':'value'}  # 另外我们也可以创建字典中的字典
d.bar.prob = 'newer'  # 另外我们也可以很方便的修改字典中元素的值
print(d)

输出:

{'foo': 3, 'bar': {'prob': 'newer'}}

3.2 在深度学习中往往利用easydict建立一个全局的变量

from easydict import EasyDict as edict
config = edict()
config.TRAIN = edict() # 创建一个字典,key是Train,值是{}
config.Test = edict()
 # config.TRAIN = {} # 这个和上面的那句话是等价的,相当于创建一个字典的key
config.TRAIN.batch_size = 25  # 然后在里面写值,表示Train里面的value也是一个字典
config.TRAIN.early_stopping_num = 10
config.TRAIN.lr = 0.0001
print(config)

输出:

{'TRAIN': {'batch_size': 25, 'early_stopping_num': 10, 'lr': 0.0001}, 'Test': {}}

参考文章

  1. easydict官方网站
  2. ipython笔记
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐