Python八种连接列表的方式

1. 最常见的相加
使用+对多个列表进行相加

>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>> list3 = [7,8,9]
>>>
>>> list1 + list2 + list3
[1, 2, 3, 4, 5, 6, 7, 8, 9]

2. 借助itertools
itertoolsPython里有一个非常强大的内置模块,它专门用于操作可迭代对象。

>>> from itertools import chain
>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>> list3 = [7,8,9]
>>>
>>> list(chain(list1, list2, list3))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

3. 使用 * 解包
使用 * 可以解包列表,解包之后再合并。

>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>>
>>> [*list1, *list2]
[1, 2, 3, 4, 5, 6]

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

>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>>
>>> list1.extend(list2)
>>> list1
[1, 2, 3, 4, 5, 6]

5. 使用列表推导式

>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>> list3 = [7,8,9]
>>>
>>> [x for l in (list1, list2, list3) for x in l]
[1, 2, 3, 4, 5, 6, 7, 8, 9]

6. 使用heapq
heapqPython的一个标准模块,它提供了堆排序算法的实现。该模块里有一个merge方法,可以用于合并多个列表。

>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>> list3 = [7,8,9]
>>>
>>> from heapq import merge
>>>
>>> list(merge(list1, list2,list3))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

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

>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>> list3 = [7,8,9]
>>>
>>> from heapq import merge
>>>
>>> list(merge(list1, list2,list3))
[1, 2, 4, 5, 3, 7, 6, 8, 9]

7. 魔法方法
有一个魔法方法叫__add__,当使用第一种方法list1+list2的时候,内部实际上是作用在__add__方法上的。

>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>>
>>> list1 + list2
[1, 2, 3, 4, 5, 6]
>>>
>>>
>>> list1.__add__(list2)
[1, 2, 3, 4, 5, 6]

借用这个特性,可以使用reduce这个方法来对多个列表进行合并。

>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>> list3 = [7,8,9]
>>>
>>> from functools import reduce
>>> reduce(list.__add__, (list1, list2,list3))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

8. 使用yield from
yield from后可接一个可迭代对象,用于迭代并返回其中的每一个元素。

>>> list1 = [1,2,3]
>>> list2 = [4,5,6]
>>> list3 = [7,8,9]
>>>
>>> def merge(*lists):
···    for l in lists:
···        yield from l
···
>>> list(merge(list1, list2, list3))
[1, 2, 3, 4, 5, 6, 7, 8, 9] 

猜你喜欢

转载自blog.csdn.net/zZZxiaozhi/article/details/115670134
今日推荐