dict.fromkeys的坑
最近需要创建一个key为字符串,value为列表的字典,key是从一个列表里获取的,所以想到了用dict.fromkeys
方法。
坑
但是,存在一个问题,fromkeys里面的第二个参数,如果传的是一个列表的话,所有的key都会指向同一个列表。而不是每个key对应一个新列表
my_list = ['one', 'two', 'three']
# 用fromkeys方法创建字典
my_dict = dict.fromkeys(my_list, [])
# 打印每个列表的id
print(id(my_dict['one']))
print(id(my_dict['two']))
print(id(my_dict['three']))
结果如下:
125553032
125553032
125553032
{'one': ['哈哈'], 'two': ['哈哈'], 'three': ['哈哈']}
上面的结果说明了,三个列表其实是一个列表,具有相同的id。
解决方法
直接不用fromkeys方法了,使用字典推导
# 使用字典推导式
my_dict = {
key:[] for key in my_dict}
# 打印每个列表的id
print(id(my_dict['one']))
print(id(my_dict['two']))
print(id(my_dict['three']))
# 给其中一个列表添加一个元素
my_dict['one'].append('哈哈')
print(my_dict)
结果如下:
125553320
125553192
21546472
{'one': ['哈哈'], 'two': [], 'three': []}
可见三个列表的id不同,是三个不同的列表,添加元素的操作也是相互独立的,不会受到影响。