在 Python 中合并列表的5种方法
共 2956字,需浏览 6分钟
·
2021-04-07 01:19
点击上方“AI算法与图像处理”,选择加"星标"或“置顶”
重磅干货,第一时间送达
引言
当我开始学习 Python 的时候,并不知道它是多么的灵活和优雅。在阅读和编写了大量代码之后,我越来越喜欢 Python。因为即使是一个普通的操作也可以有许多不同的实现。合并列表是一个很好的例子,至少有5种方法可以做到这一点。本文将介绍它们,并展示在引擎盖下的技巧。
1. 直接添加列表
在 Python 中合并列表最简单的方法就是直接使用 + 操作符,如下例所示:
leaders_1 = ['Elon Mask', 'Tim Cook']
leaders_2 = ['Yang Zhou', 'Bill Gates']
leaders_3 = ['Jeff Bezos', 'Warren Buffet']
full_leaders_list = leaders_1 + leaders_2 + leaders_3
print(full_leaders_list)
# ['Elon Mask', 'Tim Cook', 'Yang Zhou', 'Bill Gates', 'Jeff Bezos', 'Warren Buffet']
此外,+= 运算符也支持列表。但是,事情在这里变得有趣。请参见以下示例:
A = B = [1, 2, 3]
A += [4]
print(A, B)
# [1, 2, 3, 4] [1, 2, 3, 4]
A = A + [5]
print(A, B)
# [1, 2, 3, 4, 5] [1, 2, 3, 4]
上面的代码可能会令人困惑,尤其是对于 Python 初学者。
为什么第一次 A 等于 B,第二次却不等于?
在这里留下一个课堂作业,希望各位同学可以课后找到原因~
2. 扩展一个列表
除了+=运算符外,一种简单的使用列表合并的方法是使用extend()方法。
leaders_1 = ['Elon Mask', 'Tim Cook']
leaders_2 = ['Yang Zhou', 'Bill Gates']
leaders_1.extend(leaders_2)
print(leaders_1)
# ['Elon Mask', 'Tim Cook', 'Yang Zhou', 'Bill Gates']
顺便说一下,在 Python 中处理列表时,另一个名为 append ()的方法也很流行。
让我们看看如果改变前一个例子的方法会发生什么:
leaders_1 = ['Elon Mask', 'Tim Cook']
leaders_2 = ['Yang Zhou', 'Bill Gates']
leaders_1.append(leaders_2)
print(leaders_1)
# ['Elon Mask', 'Tim Cook', ['Yang Zhou', 'Bill Gates']]
如上所示,append() 将新项目添加到列表中,而extend() 将列表与其他列表连接在一起。
3. 用 Asterisks 合并列表
Python 中最美妙的技巧之一就是使用sterisks 。在asterisks 的帮助下,我们可以解压列表并将它们放在一起。这是用 Python 实现合并列表的一种令人眼花缭乱的方法(可能有点自鸣得意)。
leaders_1 = ['Elon Mask', 'Tim Cook']
leaders_2 = ['Yang Zhou', 'Bill Gates']
leaders_3 = ['Jeff Bezos', 'Warren Buffet']
full_list = [*leaders_1, *leaders_2, *leaders_3]
print(full_list)
# ['Elon Mask', 'Tim Cook', 'Yang Zhou', 'Bill Gates', 'Jeff Bezos', 'Warren Buffet']
4. 通过链函数合并列表
Itertools 模块中的 chain 函数是 Python 中合并迭代对象的一种特殊方法。它可以对一系列迭代项进行分组,并返回组合后的迭代项。因为列表也是迭代的,所以我们也可以使用 chain 函数来合并列表:
from itertools import chain
leaders_1 = ['Elon Mask', 'Tim Cook']
leaders_2 = ['Yang Zhou', 'Bill Gates']
leaders_3 = ['Jeff Bezos', 'Warren Buffet']
full_list = list(chain(leaders_1,leaders_2,leaders_3))
print(full_list)
# ['Elon Mask', 'Tim Cook', 'Yang Zhou', 'Bill Gates', 'Jeff Bezos', 'Warren Buffet']
5. 通过 Reduce 函数合并列表
Python 是懒人的福利。对我来说,当有太多的列表需要合并的时候,写太多的 + 是很无聊的,我不想这样做。在这种情况下,我们可以使用一个更高阶的函数—— reduce,再次拯救我们:
from operator import add
from functools import reduce
A = [99, 2]
B = [0, 5, 1]
C = [2077, 2021]
D = [0]
L = reduce(add, (A, B, C, D))
print(L)
# [99, 2, 0, 5, 1, 2077, 2021, 0]
总结
Python 中合并列表的操作至少有5种方法。我们不一定每次都选择不同的方式。然而,在阅读他人的程序时,不可避免地会遇到不同的编码风格。因此,对于同一个操作,检查不同的方法是值得的。至少,我们可以从他们身上感受到 Python 的灵活性和优雅。
个人微信(如果没有备注不拉群!) 请注明:地区+学校/企业+研究方向+昵称
下载1:何恺明顶会分享
在「AI算法与图像处理」公众号后台回复:何恺明,即可下载。总共有6份PDF,涉及 ResNet、Mask RCNN等经典工作的总结分析
下载2:终身受益的编程指南:Google编程风格指南
在「AI算法与图像处理」公众号后台回复:c++,即可下载。历经十年考验,最权威的编程规范!
下载3 CVPR2021 在「AI算法与图像处理」公众号后台回复:CVPR,即可下载1467篇CVPR 2020论文 和 CVPR 2021 最新论文
点亮 ,告诉大家你也在看