Python列表遍历问题：误用extend致结果异常，求原因及正确解法

我是Python完全新手，正在练习列表操作，最初运行以下代码可以正常工作：

fruits = ["apple", "banana", "cherry", "kiwi", "mango"]
newlist = []

for x in fruits:
  if "a" in x:
    newlist.append(x)

print(newlist)

上述代码运行正常

我的需求是：有两个列表，将第一个列表中包含字符"a"的元素添加到第二个列表中。但我尝试以下代码时却无法得到预期结果：

fruits = ["apple", "banana", "cherry", "kiwi", "mango"]
newlist = []

for x in fruits:
  if "a" in x:
    newlist.extend(fruits)

print(newlist)

实际输出：

['apple', 'banana', 'cherry', 'kiwi', 'mango', 'apple', 'banana', 'cherry', 'kiwi', 'mango', 'apple', 'banana', 'cherry', 'kiwi', 'mango']

** Process exited - Return Code: 0 **
Press Enter to exit terminal

预期输出：

['apple', 'banana', 'mango']

** Process exited - Return Code: 0 **
Press Enter to exit terminal

以下是对问题的分析和解决方法：

一、错误代码的执行过程 #

1. 初始化fruits列表为["apple", "banana", "cherry", "kiwi", "mango"]，同时创建空列表newlist。
2. 开始遍历fruits中的每个元素：
   • 第一个元素"apple"包含"a"，执行newlist.extend(fruits)，将整个fruits的所有元素添加到newlist，此时newlist变为['apple', 'banana', 'cherry', 'kiwi', 'mango']。
   • 第二个元素"banana"包含"a"，再次执行extend，newlist变为原列表加上整个fruits，共10个元素。
   • 第三个元素"cherry"不含"a"，跳过循环体。
   • 第四个元素"kiwi"不含"a"，跳过循环体。
   • 第五个元素"mango"包含"a"，第三次执行extend，newlist变为原列表加上整个fruits，共15个元素。
3. 最后打印newlist，得到三次重复的原列表内容。

二、错误原因 #

extend()方法的作用是将传入的可迭代对象（比如列表）的所有元素逐个追加到目标列表末尾，而不是添加单个元素。你原本想要添加当前符合条件的元素x，但错误地传入了整个fruits列表，且符合条件的元素共有3个，因此newlist被拼接了三次原列表，导致输出不符合预期。

三、正确的实现方法 #

方法1：使用append()（最直观） #

回到你最初的正确写法，每次只将符合条件的单个元素x添加到newlist：

fruits = ["apple", "banana", "cherry", "kiwi", "mango"]
newlist = []

for x in fruits:
  if "a" in x:
    newlist.append(x)

print(newlist)

方法2：列表推导式（Python风格写法） #

用一行代码完成筛选，语法更紧凑，逻辑和循环append一致：

fruits = ["apple", "banana", "cherry", "kiwi", "mango"]
newlist = [x for x in fruits if "a" in x]

print(newlist)

方法3：使用filter()函数（函数式编程风格） #

结合lambda表达式筛选元素，再转换为列表：

fruits = ["apple", "banana", "cherry", "kiwi", "mango"]
newlist = list(filter(lambda x: "a" in x, fruits))

print(newlist)

内容的提问来源于stack exchange，提问作者Aryan Singh