技術標籤：列表pythonjava程式語言人工智慧

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]
>>> 
複製程式碼

本期小編推送2021初學者一定會用到的Python資料,含有小編自己嘔心瀝血整理的免費書籍/視訊/線上文件和編輯器/原始碼,關於Ｐｙｔｈｏｎ

的安裝qun：850973621

2. 藉助 itertools

itertools 在 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. 使用 * 解包

在 Python 炫技操作(02)：合併字典的七種方法 提到了使用 ** 可解包字典。

與它相似的，使用 * 可以解包列表。 * 和 ** 常用在函式定義時，設定可變引數。

現在我將它單獨拿出來用在多個列表的合併。

示例如下：

>>> 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. 使用 heapq

heapq 是 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. 藉助魔法方法

在之前的文章裡，把魔法方法介紹得很全。

非常全的通俗易懂 Python 魔法方法指南（上）

非常全的通俗易懂 Python 魔法方法指南（下）

其中有一個魔法方法是 __add__，實際 上當我們使用第一種方法 list01 + list02 的時候，內部實際上是作用在 __add__ 這個魔法方法上的.

所以以下兩種方法其實是等價的

>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> 
>>> list01 + list02
[1, 2, 3, 4, 5, 6]
>>> 
>>> 
>>> list01.__add__(list02)
[1, 2, 3, 4, 5, 6]
>>> 
複製程式碼

借用這個魔法特性，我們可以 reduce 這個方法來對多個列表進行合併，示例程式碼如下

>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> from functools import reduce
>>> reduce(list.__add__, (list01, list02, list03))
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>>
複製程式碼

8. 使用 yield from

在很早的一篇文章裡（併發程式設計08｜深入理解yield from語法），我很詳細的介紹了 yield from 意義及使用方法。

在 yield from 後可接一個可迭代物件，用於迭代並返回其中的每一個元素。

因此，我們可以像下面這樣自定義一個合併列表的工具函式。

>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> def merge(*lists):
...   for l in lists:
...     yield from l
...
>>> list(merge(list01, list02, list03))
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>>
複製程式碼

看到這裡，有沒有漲姿勢了，學了這麼久的 Python ，沒想到合併列表還有這麼多的方法。本篇文章的主旨，並不在於讓你全部掌握這 7 種合併列表的方法，實際上，你只要選用一種最順手的方式即可。

但是在協同工作中，或者在閱讀他人程式碼時，你不可避免地會碰到各式各樣的寫法，這時候你能下意識的知道這是在做合併列表的操作，那這篇文章就是有意義的。

原文連結：https://juejin.cn/post/6844904128762609671