技术背景
其实如果没有专门去研究python的一些内置函数的话,我们都没办法发现一些很神奇的功能,即使是我们熟悉的python中的sum
函数。不知道还有多少人,以为这只是一个只能用来做求和的函数?
对列表求和
In [1]: my_list = [1,2,3,4,5] In [2]: sum(my_list) Out[2]: 15
当然,这也是sum函数的基本功能,类似于sum函数的还有min求小值函数和max求大值函数等:
In [3]: min(my_list) Out[3]: 1 In [4]: max(my_list) Out[4]: 5
当然,其实sum函数也不仅仅是可以对list这种数据结构进行求和,对tuple中的所有元素进行求和,也是可以的。
In [6]: my_tuple Out[6]: (1, 2, 3, 4, 5) In [7]: sum(my_tuple) Out[7]: 15
如果对一个字典使用sum函数,求得的结果是字典中所有键值的和:
In [8]: my_dict = {1:2,2:4,3:6,4:8,5:10} In [9]: sum(my_dict) Out[9]: 15
对异形列表进行展开
接下来是本文的重点,如果我们要对如下这样的一个异形list进行展开,我们会使用什么样的办法呢?
In [14]: my_list = [[1],[2,3],[4,5,6]]
一般我们会想到采用一个for循环来解决该问题:
In [28]: my_list_for = [] In [29]: for l in my_list: ...: my_list_for.extend(l) ...: In [30]: my_list_for Out[30]: [1, 2, 3, 4, 5, 6]
但是其实同样的操作,我们可以使用python内置的sum函数对该问题进行求解,这样在形式上就简便了很多:
In [31]: sum(my_list,[]) Out[31]: [1, 2, 3, 4, 5, 6]
非常简单,只需要一行代码,就可以实现对一个list进行展开的操作。当然,这个方法对于二维的list是可以正常使用的,可对于更加高维的嵌套异形list,使用sum函数,是得不到我们想要的效果的:
In [35]: my_list = [[1],[2,3],[4,5,[6]]] In [36]: sum(my_list,[]) Out[36]: [1, 2, 3, 4, 5, [6]]
如果是规则的list,可以多用几次sum函数即可解决,如果是异形的list,在执行的过程中很有可能报错。
In [39]: my_list = [[[1]],[[2],[3]],[[4],[5],[6]]] In [40]: sum(my_list,[]) Out[40]: [[1], [2], [3], [4], [5], [6]] In [41]: sum(sum(my_list,[]),[]) Out[41]: [1, 2, 3, 4, 5, 6]
其实相比于for循环,或者是高纬度下的while循环,这个方法只是在使用上简便一些,但是之前确实没有发现内置的sum函数还有这样的功能。
总结概要
如果我们遇到[[1],[2,3],[4,5,6]]这样的一个python列表,想把他展平为一个一维的列表,这个时候你会使用什么样的方法呢?for循环?while循环?其实python的内置函数sum中简单的支持了一下这样的功能,使得我们可以更加简便的把一个高维且长度不固定的列表展平为一个规则的一维列表。