python 字典使用

整理很好的文章
文章复制链接：

https://mp.weixin.qq.com/s/Aj65A-uuTaARW3vvYTxvzQ
1、检查键是否存在于字典中

def key_in_dict(d, key):
    return key in d

d = {'Bob': 1, 'Mary': 2, 'Lisa': 4, 'Ken': 5}
print(key_in_dict(d, 'Hello'))
print(key_in_dict(d, 'Lisa'))

输出：

False
True

2、获取字典中所有值组成的列表

这里需要使用 dict.values() 获取字典中的所有值，然后通过 list() 转换为列表。

def values_only(d):
    return list(d.values())

d = {'Bob': 1, 'Mary': 2, 'Lisa': 4, 'Ken': 5}
print(values_only(d))

输出：

[1, 4, 2, 5]

3、获取字典中所有键组成的列表

这里需要使用 dict.keys() 获取字典中的所有键，然后通过 list() 转换为列表。

def keys_only(d):
    return list(d.keys())

d = {'Bob': 1, 'Mary': 2, 'Lisa': 4, 'Ken': 5}
print(keys_only(d))

输出：

['Lisa', 'Mary', 'Ken', 'Bob']

4、将字典所有键值对转换为元组列表

这里需要使用 dict.items() 获取字典中的所有键值对，然后通过 list() 转换为元祖列表。

def dict_to_list(d):
    return list(d.items())

d = {'Bob': 1, 'Mary': 2, 'Lisa': 4, 'Ken': 5}
print(dict_to_list(d))

输出：

[('Ken', 5), ('Lisa', 4), ('Bob', 1), ('Mary', 2)]

5、将字典中的键值进行互换

这里仍然需要借助 dict.items() 分别获取字典的键和值，互换位置，来创建一个新字典。

def invert_dictionary(d):
    return { value: key for key, value in d.items() }

d = {'Bob': 1, 'Mary': 2, 'Lisa': 4, 'Ken': 5}
print(invert_dictionary(d))

输出：

{1: 'Bob', 2: 'Mary', 4: 'Lisa', 5: 'Ken'}

6、使用 dict.get(key)，而不是 dict[key]

通过键获取对应的值，dict[key] 是最常用的方法，但是如果 key 不存在，则会引发 KeyError 错误。

那么你可以使用 dict.get(key) 来规避这个问题，dict.get() 可以接受第二个参数，表示该键对应的默认值，当 key 不存在时，就返回该默认值。

d = {'Bob': 1, 'Mary': 2, 'Lisa': 4, 'Ken': 5}
print(d.get('Mary'))
print(d.get('mary', 2))

print(d['mary'])  # KeyError

输出：

2
2

7、将字典列表转换为指定键对应的值列表

如果你有多个字典，且字典的键都相同，则可以根据指定键，获取每个字典该键的值组成的列表。

需要使用到 dict.get() 来得到键对应的值。

def get_values_by_key(dicts, key):
  return [x.get(key) for x in dicts]
students = [
  { 'name': 'Bob',  'id': 10 },
  { 'name': 'Mary', 'id': 20 },
  { 'name': 'Lisa', 'id': 30 },
  { 'name': 'Ken',  'id': 40 }
]
print(get_values_by_key(students, 'name'))
print(get_values_by_key(students, 'id'))

输出：

['Bob', 'Mary', 'Lisa', 'Ken']
[10, 20, 30, 40]

8、查找字典中最小值对应的键

使用 min() 函数，同时设置参数 key 为 dict.get()。同理，要获取最大值，则使用 max()。

def key_of_min(d):
    return min(d, key = d.get)

d = {'Bob': 1, 'Mary': 2, 'Lisa': 4, 'Ken': 5}
print(key_of_min(d))

输出：

Bob

9、合并多个字典

创建一个新字典，并使用 dict.update() 将传入的每个字典添加到新字典中。

def merge_dicts(*dicts):
    res = dict()
    for d in dicts:
        res.update(d)
    return res

stud1 = {'Bob': 1, 'Mary': 2}
stud2 = {'Lisa': 4, 'Ken': 5}
stud3 = {'Vivi': 7}

print(merge_dicts(stud1, stud2, stud3))
输出：

{'Lisa': 4, 'Vivi': 7, 'Mary': 2, 'Ken': 5, 'Bob': 1}

10、根据值查找对应的键

使用 dict.items() 和 next() 找到字典中具有给定值的第一个键。

如果想找到字典中具有给定值的所有键，则将 next() 更改为 list()。

def find_key(d, val):
    return next(key for key, value in d.items() if value == val)

d = {'Bob': 1, 'Mary': 2, 'Lisa': 4, 'Ken': 5, 'Vivi': 2}
print(find_key(d, 2))

输出：

Mary

11、将两个列表组合为一个字典

这里有两个列表，第一个列表存放键，第二个列表存放值，要将这两个列表转换为一个字典。

def lists_to_dict(keys, values):
    return dict(zip(keys, values))

names = ['Bob', 'Mary', 'Lisa', 'Ken']
ids = [1, 2, 4, 5]

print(lists_to_dict(names, ids))
输出：

{'Ken': 5, 'Lisa': 4, 'Bob': 1, 'Mary': 2}

12、根据给定的函数对列表中的元素进行分组

对列表中的每个元素执行相同的操作，并根据结果对元素进行分组。

需要使用 collections.defaultdict 初始化一个特殊字典，其中的值都是列表。

下面例子是根据每个名字的长度进行分组。

from collections import defaultdict

def group_by(items, fn):
    d = defaultdict(list)
    for el in items:
        d[fn(el)].append(el)
    return dict(d)

print(group_by(['Bob', 'Mary', 'Lisa'], len))

输出：

{3: ['Bob'], 4: ['Mary', 'Lisa']}