元组拆包
元组是不可变列表,列表是通过索引取值的,元组也是:
tuple_test = (1, 2, 3) a = tuple_test[0] b = tuple_test[1] c = tuple_test[2]
但Python是出了名的一行代码解决问题,元组拆包就是精髓技术之一:
a, b, c = tuple_test
print("%s %s %s" % tuple_test)
把元组一一对应拆出来,就叫做元组拆包。拆包有个要求,元组中的元素数量必须跟接受这些元素的空挡数一致,否则会报错:
tuple_test = (1, 2, 3) a, b = tuple_test # ValueError: too many values to unpack (expected 2)
_占位符
使用_占位符可以解决这个问题:
tuple_test = (1, 2, 3) a, b, _ = tuple_test
这样就只获取到部分数据了,这在取函数返回值时特别有用,比如:
import os _, filename = os.path.split("/home/dongfanger/.ssh/idrsa.pub") print(filename) # "idrsa.pub"
*前缀
当返回值特别多时,_占位符写起来麻烦,可以用*来处理剩下的元素:
>>> a, b, *rest = range(5) >>> a, b, *rest (0, 1, [2, 3, 4])
注意rest是个列表,如果没有足够元素,会返回空列表:
>>> a, b, *rest = range(2) >>> a, b, *rest (0, 1, [])
*前缀变量能放在任意位置,比如,放在中间:
>>> a, *body, c, d = range(5) >>> a, body, c, d (0, [1, 2], 3, 4)
放在前面:
>>> *head, b, c, d = range(5) >>> head, b, c, d ([0, 1], 2, 3, 4)
实在是妙啊。
*还有一个作用,把元组拆开作为函数参数:
>>> divmod(20, 8) (2, 4) >>> t = (20, 8) >>> divmod(*t) (2, 4)
经典写法*args就是这个道理。
嵌套元组拆包
嵌套元组是指元组中有元组,比如(1, 2, 3, (4, 5)),对于嵌套元组,你可能会想要拆两遍:
tuple_nest_test = (1, 2, 3, (4, 5)) a, b, c, d = tuple_nest_test x, y = d print(a, b, c, x, y)
实际上能一步到位:
tuple_nest_test = (1, 2, 3, (4, 5)) a, b, c, (x, y) = tuple_nest_test print(a, b, c, x, y)
交换两个变量的值
元组拆包提供了语法糖,对于交换两个变量的值的常规写法:
temp = a a = b b = temp
可以切换为优雅写法:
b, a = a, b
具名元组
元组很像数据库表记录,除了没有表名和字段名,collections.namedtuple具名元组补偿了这个缺憾,它是一个工厂函数,可以用来构建一个带字段名的元组和一个有名字的类,比如:
import collections # 定义 Card = collections.namedtuple("Card", ["rank", "suit"]) # 初始化 card_test = Card("J", "hearts") # 使用 print(card_test.rank) # J print(card_test[1]) # hearts
Card是表名,有两个表字段rank和suit。
定义具名元组需要2个参数,第1个参数是类名,第2个参数是字段名,既可以是可迭代对象(如列表和元组),也可以是空格间隔的字符串:
Card = collections.namedtuple("Card", ("rank", "suit")) Card = collections.namedtuple("Card", "rank suit")
初始化时以一串参数形式传入构造函数:
card_test = Card("J", "hearts")
既可以通过.运算符,也可以用索引来取值:
print(card_test.rank) print(card_test[1])
这个带名字的元组,对调试程序有很大帮助。
列表与元组
元组是不可变列表,它们就像双胞胎,长相类似,内在性格却有不同:
黄色列表独有,红色元组特有,元组竟然还多了个s.__getnewargs__()方法!从表中可以清楚地看到,除了跟增减元素相关的方法之外,元组支持列表的其他所有方法。
列表也能拆
既然列表和元组是孪生兄弟,那必然也有共同技能:
list_test = [1, 2, 3] a, b, c = list_test
>>> divmod(20, 8) (2, 4) >>> t = [20, 8] # 换成列表 >>> divmod(*t) (2, 4)
列表拆包,也是ok的。
小结
本文介绍了Python神奇操作元组拆包,借助_占位符和*前缀可以进行更加灵活的取值,具名元组实际用的还比较少,不过看一些源码是有的。文章最后比较了列表和元组的差异,列表也能拆包。列表(list)、元组(tuple),以及字符串(str),都有一个共同操作:切片。