python爬虫
直播中

mushenmu

3年用户 732经验值
擅长:可编程逻辑
私信 关注
[经验]

python字典高阶用法

通过序列生成字典
我们将下面的序列转换为dict类型。
  1. lst = [('a', 1), ('b', 2), ('c', 3)]
普通的写法
  1. for k, v in lst:
  2.     dic[k] = v
更pythonic的写法
利用字典推导式快速生成字典。
  1. {k: v for k, v in lst}
key的默认值
当指定key不存在时,将value设置为 0。
普通的写法
  1. if key not in dct:
  2.     dct[key] = 0
pythonic的写法
  1. dct[key] = dct.get(key, 0)
交换key与value
普通的写法
  1. dic = {'Python': 1, 'Java': 2}
  2. new_dic = {}
  3. for k, v in dic.items():
  4.     new_dic[v] = k
更pythonic的写法
  1. dic = {'Python': 1, 'Java': 2}
  2. new_dic = {v: k for k, v in dic.items()}
序列修改和初始化
示例数据
  1. lst = [('a', 1), ('b', 2), ('c', 3)]
  2. dic = {'a': [0]}
如果我们需要根据lst来更新dic中的数据,当key存在,则将value添加到原序列末尾,否则初始化value并用序列保存。
普通的写法
  1. for key, value in lst:
  2.     if key in dic:
  3.         dic[key].append(value)
  4.     else:
  5.         dic[key] = [value]
更pythonic的写法
  1. for (key, value) in lst:
  2.     group = dic.setdefault(key, [])
  3.     group.append(value)
  4. # dic:{'a': [0, 1], 'b': [2], 'c': [3]}
setdefault(key, default)会先判断key是否存在,存在则返回dct[key] , 不存在则把dct[key]设为 [] 并返回。
key,items的集合运算
如果我们现在需要获取两个字典的key相交的部分的映射信息。
普通的写法
  1. dic1 = {'Python': 1, 'Java': 2, 'C': 3}
  2. dic2 = {'Python': 3, 'Java': 2, 'C++': 1}

  3. new_dic = {}
  4. for k, v in dic1.items():
  5.     if k in dic2.keys():
  6.         new_dic[k] = v
  7. print(new_dic)

  8. # {'Python': 1, 'Java': 2}
更pythonic的写法
  1. dic1 = {'Python': 1, 'Java': 2, 'C': 3}
  2. dic2 = {'Python': 3, 'Java': 2, 'C++': 1}

  3. print({k: dic1[k] for k in dic1.keys() & dic2.keys()})

  4. # {'Python': 1, 'Java': 2}
这里的dic1.keys() & dic2.keys()用到的就是 keys()进行集合运算,items()同样可以进行集合运算。
如果现在我们要获取两个字典中 key,value 完全相同的部分。
  1. dic1 = {'Python': 1, 'Java': 2, 'C': 3}
  2. dic2 = {'Python': 3, 'Java': 2, 'C++': 1}

  3. print(dic1.items() & dic2.items())

  4. # {('Java', 2)}
灵活运用 keys,items() 集合运算的特性,可以快速提取我们想要的内容。
按key或value对字典排序
使用sorted()函数快速实现对key或value的排序。
  1. dic = {'a': 2, 'b': 1, 'c': 3, 'd': 0}
  2. lst1 = sorted(dic.items(), key=lambda x: x[0], reverse=False)
  3. # [('a', 2), ('b', 1), ('c', 3), ('d', 0)]
  4. lst2 = sorted(dic.items(), key=lambda x: x[1], reverse=False)
  5. # [('d', 0), ('b', 1), ('a', 2), ('c', 3)]
  6. print('按照键降序:', {key: value for key, value in lst1})
  7. print('按照值降序:', {key: value for key, value in lst2})

  8. # 按照键降序: {'a': 2, 'b': 1, 'c': 3, 'd': 0}
  9. # 按照值降序: {'d': 0, 'b': 1, 'a': 2, 'c': 3}
多个字典排序
如果一个序列中包含多个字典,现在要根据条件对这些字典继续排序。同样可以使用sorted()函数来实现。
d
  1. ict_list = [
  2.     {'letter': 'B', 'number': '2'},
  3.     {'letter': 'A', 'number': '3'},
  4.     {'letter': 'B', 'number': '1'}
  5. ]

  6. # 按 letter 排序
  7. print(sorted(dict_list,
  8.              key=lambda dic: dic['letter']))
  9. # 按 letter, number 排序
  10. print(sorted(dict_list,
  11.              key=lambda dic: (dic['letter'], dic['number'])))
  12.             
  13. # [{'letter': 'A', 'number': '3'}, {'letter': 'B', 'number': '2'}, {'letter': 'B', 'number': '1'}]
  14. # [{'letter': 'A', 'number': '3'}, {'letter': 'B', 'number': '1'}, {'letter': 'B', 'number': '2'}]
当然,如果你知道itemgetter()的话,上面的代码就可以改变一下,执行速度会更快。
  1. from operator import itemgetter

  2. print(sorted(dict_list,
  3.              key=itemgetter('letter')))
  4. print(sorted(dict_list,
  5.              key=itemgetter('letter', 'number')))
itemgetter()获取的不是值,而是定义了一个函数,通过该函数作用到目标对象上。

更多回帖

发帖
×
20
完善资料,
赚取积分