[经验]

python八种连接列表的方式

mushenmu

2022-4-7 17:04:54

1785 python

1、最直观的相加使用 + 对多个列表进行相加，你应该懂，不多说了。

>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> list01 + list02 + list03
[1, 2, 3, 4, 5, 6, 7, 8, 9]

复制代码

2、借助 itertoolsitertools 在 Python 里有一个非常强大的内置模块，它专门用于操作可迭代对象。
在前面的文章中也介绍过，使用 itertools.chain() 函数先将可迭代对象（在这里指的是列表）串联起来，组成一个更大的可迭代对象。
最后你再利用 list 将其转化为列表。

>>> from itertools import chain
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> list(chain(list01, list02, list03))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

复制代码

3、使用 * 解包使用 * 可以解包列表，解包后再合并。
示例如下：

>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>>
>>> [*list01, *list02]
[1, 2, 3, 4, 5, 6]

复制代码

4、使用 extend在字典中，使用 update 可实现原地更新，而在列表中，使用 extend 可实现列表的自我扩展。

>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>>
>>> list01.extend(list02)
>>> list01
[1, 2, 3, 4, 5, 6]

复制代码

5、使用列表推导式Python 里对于生成列表、集合、字典，有一套非常 Pythonnic 的写法。
那就是列表解析式，集合解析式和字典解析式，通常是 Python 发烧友的最爱，那么今天的主题：列表合并，列表推导式还能否胜任呢？
当然可以，具体示例代码如下：

>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> [x for l in (list01, list02, list03) for x in l]
[1, 2, 3, 4, 5, 6, 7, 8, 9]

复制代码

6、使用 heapqheapq 是 Python 的一个标准模块，它提供了堆排序算法的实现。
该模块里有一个 merge 方法，可以用于合并多个列表，如下所示

>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> from heapq import merge
>>>
>>> list(merge(list01, list02, list03))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

复制代码

要注意的是，heapq.merge 除了合并多个列表外，它还会将合并后的最终的列表进行排序。

>>> list01 = [2,5,3]
>>> list02 = [1,4,6]
>>> list03 = [7,9,8]
>>>
>>> from heapq import merge
>>>
>>> list(merge(list01, list02, list03))
[1, 2, 4, 5, 3, 6, 7, 9, 8]

复制代码

它的效果等价于下面这行代码：

sorted(itertools.chain(*iterables))

复制代码

如果你希望得到一个始终有序的列表，那请第一时间想到 heapq.merge，因为它采用堆排序，效率非常高。但若你不希望得到一个排过序的列表，就不要使用它了。
7、借助魔法方法有一个魔法方法叫 __add__，当我们使用第一种方法 list01 + list02 的时候，内部实际上是作用在 __add__ 这个魔法方法上的。
所以以下两种方法其实是等价的