python 列表删除多个元素

文章目录

      • 一. 删除列表多个元素
        • 方法1 使用枚举:
        • 2. 使用python中List.pop()方法
        • 3. 使用python中List.remove()方法
        • 4. 注意
      • 二. 使用双指针法删除列表多个元素
        • 1. 问题描述:
        • 2. 解决方法:
        • 3. 代码如下:
      • 三. 总结
      • 四. 相关链接

一. 删除列表多个元素

问题背景:
删除列表中的元素,其所在的位置为[1,3,7]

a=[1,4,2,1,5,6,9,0]

期望输出结果:

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

方法1 使用枚举:

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

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

完整代码:

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}")

2. 使用python中List.pop()方法

使用内置方法remove移除,但是每次只能移除一个:

#pop
L.pop([index]) -> item -- remove and return item at index (default last).#(从后往前,有返回值)
#输入为:元素所在的索引
  1. 若直接使用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.pop(i)# b.pop(i)
      7 print(b)
      8 

IndexError: list index out of range
  1. 改进:
b=a.copy()
for num,i in enumerate(del_index):
    print(f"{b},被移除的元素:{a[i]}")
    b.pop(i-num)
print(f"最终结果:{b}")

结果与枚举法结果一致。

3. 使用python中List.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进行移除。虽然得到的结果值一样,但是元素的索引与其它方法的结果不一致。

4. 注意

  • 就最终结果而言,枚举法和pop法不受重复值的影响;而remove容易受重复值的影响。若最终结果不受顺序影响,三者的差异可以忽略不计。
  • 推荐使用枚举法删除列表多个元素,pop()方法和remove()方法删除列表元素容易出错,不推荐使用,因为删除列表元素后,列表大小会改变,从而导致列表的索引也会改变,从而导致出错。
  • 如果只删除列表一个元素,可以使用列表的pop()和remove()方法。

二. 使用双指针法删除列表多个元素

使用双指针法删除列表多个元素解决方法链接

1. 问题描述:

LeetCode链接
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。

不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并原地修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1: 给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。

示例 2: 给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。

2. 解决方法:

双指针法(快慢指针法): 通过一个快指针和慢指针在一个for循环下完成两个for循环的工作
定义快慢指针:

  • 快指针:寻找新数组的元素 ,新数组就是不含有目标元素的数组
  • 慢指针:指向更新 新数组下标的位置
    删除过程如下:
    删除

注意双指针方法并没有改变元素的相对位置(就地删除列表元素):

  • 时间复杂度:O(n)
  • 空间复杂度:O(1)

3. 代码如下:

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        # 快慢指针
        fast = 0  # 快指针
        slow = 0  # 慢指针
        size = len(nums)
        while fast < size:  # 不加等于是因为,a = size 时,nums[a] 会越界
            # slow 用来收集不等于 val 的值,如果 fast 对应值不等于 val,则把它与 slow 替换
            if nums[fast] != val:
                nums[slow] = nums[fast]
                slow += 1
            fast += 1
        return slow

三. 总结

  • 推荐使用枚举法和双指针方法删除列表多个元素,不推荐使用列表的pop()方法和remove()方法,容易出错。
  • 如果只删除列表一个元素,可以使用列表的pop()和remove()方法。

四. 相关链接

  1. Python 列表中多元素的删除(移除)
  2. 使用双指针法删除列表多个元素解决方法链接

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2023年4月29日
下一篇 2023年4月29日

相关推荐