Python列表中多元素删除(移除)的实现

2023-03-10 11:15

短信预约 -IT技能 免费直播动态提醒

问题背景

a=[1,4,2,1,5,6,9,0]
#删除列表中的元素,其所在的位置为[1,3,7]
del_index=[1,3,7]
tmp=[i for num,i in enumerate(a) if num not in del_index]
print(f"最终结果：{tmp}")

结果：

最终结果：[1, 2, 5, 6, 9]

目的是实现List中的多个元素的移除(单个元素的比较简单，这里不做过多赘述)。而多元素的删除，可以归纳为以下两类：

实现方法

1. 使用枚举法

直接使用枚举，然后判断索引在不在删除索引里面即可。

[i for num,i in enumerate(a) if num not in del_index]

2. python中List的内置方法

使用内置方法remove移除或pop，但是每次只能移除一个。具体的函数差异：

#pop
L.pop([index]) -> item -- remove and return item at index (default last).#(从后往前，有返回值)
#输入为：元素所在的索引

#remove:  
 L.remove(value) -> None -- remove first occurrence of value.
#(无返回值)
#输入为具体的元素,重复的元素首先移除第一个

若直接使用for循环直接使用上述函数，就会报错！！！比如：

--------------------------------------------------------
IndexError Traceback (most recent call last)
<ipython-input-25-364cdd187e0c> in <module>
4 b=a.copy()
5 for i in del_index:
----> 6 b.remove(b[i])# b.pop(i)
7 print(b)
8

IndexError: list index out of range

改进：

(1) pop

b=a.copy()
for num,i in enumerate(del_index):
    print(f"{b},被移除的元素：{a[i]}")
    b.pop(i-num)
print(f"最终结果：{b}")

结果与枚举法结果一致。

(2) remove

c=a.copy()
for num,i in enumerate(del_index):
    print(f"{c},被移除的元素：{a[i]}")
    c.remove(c[i-num])#重复元素
print(f"最终结果：{c}")

输出结果：

[1, 4, 2, 1, 5, 6, 9, 0],被移除的元素：4
[1, 2, 1, 5, 6, 9, 0],被移除的元素：1
[2, 1, 5, 6, 9, 0],被移除的元素：0
最终结果：[2, 1, 5, 6, 9]

remove后的结果与上述方法的结果不一致.主要原因是重复值的问题，a[3]=1,a[0]=1,所以在移除元素4后，原列表变成[1, 2, 1, 5, 6, 9, 0].移除元素1时，本应该移除的是4右侧的1，但是由于remove的默认属性，它会选择第一次出现的1进行移除。虽然得到的结果值一样，但是元素的索引与其它方法的结果不一致。