Python中extend的使用方法

Python 笔记

一、extend()方法简介

Python中的list类型有一个重要的方法叫做extend()，可以在列表的末尾添加另一个列表的元素。

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

可以看到，list1现在包含了原本的三个元素和list2的三个元素。

二、extend()与append()的区别

extend()方法和append()方法都可以向列表中添加元素，但是两者有一个重要的区别。

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

可以看出，使用append()方法时，list2会作为一个元素添加到list1的末尾，而使用extend()方法时，则是将list2中的元素依次添加到list1的末尾。

三、使用extend()方法进行快速连接

extend()方法还有一个重要的应用场景，就是可以使用它来快速连接多个列表。

>>> list1 = [1, 2, 3]
>>> list2 = [4, 5, 6]
>>> list3 = [7, 8, 9]
>>> result = []
>>> result.extend(list1)
>>> result.extend(list2)
>>> result.extend(list3)
>>> print(result)
[1, 2, 3, 4, 5, 6, 7, 8, 9]

可以看到，我们可以通过多次调用extend()方法，将多个列表连接在一起。

四、使用*运算符进行列表扩展

除了使用extend()方法外，我们还可以使用列表的*运算符来进行扩展，类似于字符串的重复操作。

>>> list1 = [1, 2, 3]
>>> list2 = [4, 5, 6]
>>> combined = list1 + list2
>>> multiplied = list1 * 3
>>> print(combined)
[1, 2, 3, 4, 5, 6]
>>> print(multiplied)
[1, 2, 3, 1, 2, 3, 1, 2, 3]

可以看到，使用+运算符可以将两个列表合并成一个，而使用*运算符可以将一个列表重复多次。

五、使用extend()方法移除嵌套列表

除了可以用extend()方法来添加多个元素，还可以用它来从列表中移除嵌套的列表。

>>> list1 = [[1, 2], [3, 4], [5, 6]]
>>> flattened = []
>>> for sublist in list1:
>>>     flattened.extend(sublist)
>>> print(flattened)
[1, 2, 3, 4, 5, 6]

可以看到，我们可以用for循环来遍历嵌套的列表，然后通过extend()方法将子列表中的元素一个一个添加到新的列表中。

六、总结

通过本文的介绍，我们了解了Python中list类型的extend()方法的使用方法以及与append()方法的区别，还学习了如何通过extend()方法快速连接多个列表、如何使用*运算符进行列表扩展、以及如何使用extend()方法移除嵌套的列表。掌握了这些知识，我们可以更加便捷地处理列表。