Python的字典(dictionary)是一种灵活的数据结构类型,字典的每个键值对(key=>value)用冒号(:)分割,每个对之间用逗号(,)分割。Python字典里的键必须独一无二,但值则不必的。字典的值可以取任何数据类型,但必须是不可变的(unhashable),如字符串,元组或数值, 用列表是不行的。本文教你一文看懂Python字典类型数据常见操作和排序。
字典的创建
字典的创建主要有2种方法: 直接赋值和根据键赋值。下例中分别使用2种方法创建了字典d1和字典d2。
>>> d1 = {'key3': 4, 'key2': 5, 'key1': 4}
>>> print(d1)
{'key3': 4, 'key2': 5, 'key1': 4, }
>>> d2 = dict()
>>> d2['key5']=1
>>> print(d2)
{'key5': 1}
访问字典的键与值
你可以直接根据dict['键名']获取一个值,也可以通过dict.keys()获取所有的键或者dict.values()获取所有的值。但需要注意的是dict.keys()或dict.values()返回的并非列表,你需要使用list方法才能将所有的键和值转化为列表形式。
>>> d1 = {'key3': 4, 'key2': 5, 'key1': 4}
>>> print(d1['key3'])
4
>>> print(d1.keys())
dict_keys(['key3', 'key2', 'key1'])
>>> print(list(d1.keys()))
['key3', 'key2', 'key1']
>>> print(list(d1.values()))
[4, 5, 4]
注意: 当你尝试用一个不存在的键获取值时会出现KeyError,一个更好的习惯是在获取值前先使用if key in dict来判断一个键是否已经存在。
>>> print(d1['key6'])
Traceback (most recent call last):
File "<input>", line 1, in <module>
KeyError: 'key6'
>>> print('key6' in d1)
False
遍历字典
使用for key, value in dict.items()可以遍历整个字典。如果你只喜欢遍历键名或值,你可以只使用for key in dict.keys()或 for value in dict.values()。
for key, value in d1.items():
print("{}->{}".format(key, value))
for key in d1.keys():
print(key)
for value in d1.values():
print(value)
字典数据的添加
字典与列表一样都是可变类型的数据,所以可以实现数据的添加和修改。实现的方法主要有两种:1是创建新的键名并赋值,2是使用update方法。dict自带的update方法非常有用,不仅可以插入新的键值对,还是实现两个字典合并的首选方法。
>>> print(d1)
{'key3': 4, 'key2': 5, 'key1': 4}
>>> d1.update({'key4':2})
>>> print(d1)
{'key3': 4, 'key2': 5, 'key1': 4, 'key4': 2}
>>> d1['key5']=1
>>> print(d1)
{'key3': 4, 'key2': 5, 'key1': 4, 'key4': 2, 'key5': 1}
注意: 当创建的键名已存在时,dict['键名']只会更新现有值,而不会创建新的键值对。
删除字典元素
利用del dict['key']可以删除一个键值对,利用del dict可以删除整个字典, 使用dict.clear()方法可以清空一个字典。三者还是有很大区别的。
>>> print(d1)
{'key3': 4, 'key2': 5, 'key1': 4, 'key4': 2, 'key5': 1}
>>> del d1['key5']
>>> print(d1)
{'key3': 4, 'key2': 5, 'key1': 4, 'key4': 2}
>>> d1.clear()
>>> print(d1)
{}
>>> del d1
>>> print(d1)
Traceback (most recent call last):
File "<input>", line 1, in <module>
NameError: name 'd1' is not defined
利用zip方法由两个列表创建字典
zip()函数来可以把2个或多个列表合并,并创建一个元组对的列表,使用dict方法可以将其变成字典。元组对的数量以合并列表的最短长度为准。python 3中zip方法合并列表后生成的是zip对象,必需使用dict方法才能将其变成字典。
>>> l1 = [ 1, 2, 3 ]
>>> l2 = [ 'x', 'y', 'z']
>>> l3 = [ 'x', 'y' ]
>>> zip(l1, l2)
<zip object at 0x031D6828>
>>> print(list(zip(l1, l2)))
[(1, 'x'), (2, 'y'), (3, 'z')]
>>> print(list(zip(l1, l3)))
[(1, 'x'), (2, 'y')]
>>> print(dict(zip(l1,l3)))
{1: 'x', 2: 'y'}
实际上zip方法支持所有可迭代对象(字符串、列表、元祖、字典), 而不仅仅是列表。利用这个特性,我们可以很容易创建各种字典,包括很复杂的字典。我们来看下面2个经典例子。注意zip对象支持直接遍历哦,不需要先转成list或dict哦。
>> > l1 = [1, 2, 3]
>> > str1 = "abc"
>> > print(dict(zip(l1, str1)))
{1: 'a', 2: 'b', 3: 'c'}
>> > name = ["John", "Jim", "Lucy"]
>> > year = [1983, 1985, 1995]
>> > birth_year = dict(zip(name, year))
>> > print(birth_year)
{'John': 1983, 'Jim': 1985, 'Lucy': 1995}
>> > for name, year in zip(name, year):
print("{} - {}".format(name, year))
John - 1983
Jim - 1985
Lucy - 1995
利用zip方法实现键值反转
前文提到zip方法支持所有可迭代对象,也自然支持dict.values()和dict.keys(),利用这个特性我们可以快速实现一个字典的键值反转,如下所示:
>>> d1 = {'key3': 4, 'key2': 5, 'key1': 4, 'key4': 2, }
>>> d2 = dict(zip(d1.values(), d1.keys()))
>>> print(d2)
{4: 'key1', 5: 'key2', 2: 'key4'}
为什么使用zip方法后字典少了一个键? 还记得我们之前提到过键名必需唯一吗? 可见zip方法对字典数据进行反转必需考虑值相同而导致数据丢失的问题。
单个字典根据键名或值排序
使用sorted方法可以对单个字典的键名进行正向或逆向排序(reverse=True),得到是一个排过序的键名列表。然后我们通过使用for key in sorted(d)遍历排过序的键从而得到排过序的值。
>>> d1 = {'key3': 4, 'key2': 5, 'key1': 4, 'key4': 2, }
>>> d3 = sorted(d1)
>>> print(d3)
['key1', 'key2', 'key3', 'key4']
>>> print(sorted(d1, reverse=True))
['key4', 'key3', 'key2', 'key1']
>>> for key in sorted(d1):
print(d1[key])
4
5
4
2
如果我们希望根据值来排序,我们一般有2种方法: 1. 利用zip方法构建键值反转字典再排序。2. 使用匿名函数lambda。方法2通常是一种更好的方法。
方法1: 使用zip方法反转字典
>>> d1 = {'key3': 4, 'key2': 5, 'key1': 4, 'key4': 2, }
>>> d2 = dict(zip(d1.values(), d1.keys()))
>>> print(list(d2.keys()))
[4, 5, 2]
>>> for key in sorted(d2):
print(d2[key])
key4
key1
key2
由于使用zip方法导致了数值的缺失,一个更好的方法是使用lambda函数指定key来排序。
方法2: 使用lambda匿名函数
该方法设置排序的key为d1[k], 即每个键值对的值来排序。
>>> d1 = {'key3': 4, 'key2': 5, 'key1': 4, 'key4': 2, }
>>> d2 = sorted(d1, key=lambda k: d1[k])
>>> print(d2)
['key4', 'key3', 'key1', 'key2']
>>> for key in d2:
print(d1[key])
2
4
4
5
字典的运算
假设我们有如下两个字典,我们能实现直接相加或相减吗? 答案是不能。Python 3并不支持字典的直接相加或相减。如果你要合并两个字典,需要使用dict.update()方法。
>>> d1 = {'key1':1, 'key2':7}
>>> d2 = {'key1':3, 'key3':4}
虽然python不支持加减运算,但支持针对keys()和values()类似集合的运算, 可以用来求同求异,代码如下所示:
>>> d1.keys() & d2.keys()
{'key1'}
>>> d1.keys() - d2.keys()
{'key2'}
你有没有发现字典的默认运算一般是针对键进行的?
提取字典子集
假设我们有如下一个字典(姓名->分数),我们需要提取分数大于60的键值对组成一个新的字典子集,我们应该怎么办呢?
d = {'John':50, 'Mary': 65, 'Kitty':90, 'Tom': 100}
最好最快的方式就是使用字典推倒表达式,方法如下。该方法工作常用,考试必考,请一定要熟练掌握哦。
>>> d = {'John':50, 'Mary': 65, 'Kitty':90, 'Tom': 100}
>>> sub_d = { key:value for key, value in d.items() if value >= 60 }
>>> print(sub_d)
{'Mary': 65, 'Kitty': 90, 'Tom': 100}
字典列表的排序
实际应用中更常见的是对字典列表排序,而不是对当个字典的键或值进行排序。假设你有一个字典列表,你想根据某个或某几个字典字段来排序这个列表。这时你需要使用 operator
模块的 itemgetter
函数,可以非常容易的排序这样的数据结构。 假设你从数据库中检索出来网站会员信息列表,并且以下列的数据结构返回:
rows = [
{'fname': 'Brian', 'lname': 'Jones', 'uid': 1003},
{'fname': 'David', 'lname': 'Beazley', 'uid': 1002},
{'fname': 'John', 'lname': 'Cleese', 'uid': 1001},
{'fname': 'Big', 'lname': 'Jones', 'uid': 1004}]
根据任意的字典字段来排序输入结果行是很容易实现的,代码示例:
from operator import itemgetter
rows_by_fname = sorted(rows, key=itemgetter('fname'))
print(rows_by_fname)
代码的输出如下。注意该方法返回的直接排过序的完整字典,而不是排过序的键名列表。
[{'fname': 'Big', 'uid': 1004, 'lname': 'Jones'},
{'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'},
{'fname': 'David', 'uid': 1002, 'lname': 'Beazley'},
{'fname': 'John', 'uid': 1001, 'lname': 'Cleese'}]
itemgetter()
函数也支持多个 keys,比如下面的代码
rows_by_lfname = sorted(rows, key=itemgetter('lname','fname'))
print(rows_by_lfname)
输出如下:
[{'fname': 'David', 'uid': 1002, 'lname': 'Beazley'},
{'fname': 'John', 'uid': 1001, 'lname': 'Cleese'},
{'fname': 'Big', 'uid': 1004, 'lname': 'Jones'},
{'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'}]
本节关于字典列表的排序介绍来自python cookbook。
字典列表的去重
假设我们有如下一个字典列表,我们需要实现字典值的去重,即提取所有独特的值组成一个列表。一个最好的方式就是使用列表表达式。
>>> d1 = [{'key1':2}, {'key1':3}, {'key2': 2}, {'key3':4}]
>>> u_value = set(val for dict in d1 for val in dict.values())
>>> print(list(u_value))
[2, 3, 4]
同样的你可以使用列表表达式提取所有独特的键名组成一个列表。
>>> d1 = [{'key1':2}, {'key1':3}, {'key2': 2}, {'key3':4}]
>>> u_key = set(key for dict in d1 for key in dict.keys())
>>> print(list(u_key))
['key2', 'key3', 'key1']
小结
本文总结了Python字典类型数据常见操作(如创建,添加,修改和删除), 还重点讲解了字典和字典列表的排序和去重。如果喜欢本文就点赞吧。