Python实现列表索引批量删除的5种方法

(编辑：jimmy 日期: 2024/11/2 浏览：3 次 )

最近用Java做项目，使用到List broadcastInfos的 broadcastInfos.remove()方法，出现项目的重大失误，因为第一次使用Java语言，过于相信remove()方法，所以，不加思索的就使用了来删除list对象中的指定元素。

背景：
目标对象 listObj：[3, 4, 5, 6]
删除指定索引列表 indexList： [1, 2]
返回结果： [3, 6]
常见错误：

for listElement in listObj:
	for index in indexList:
		if listElement == listObj[index]:
			listObj.remove(listElement)

返回结果

这种结果，既不会报错，返回值也与目标极为相似，返回2个值list对象，len(listObj)-2 = 2，也是[3,… 开头，如果不仔细看好像没什么问题。但是结果却是错误的。

将 listObj.remove() 改为listObj.pop(index)结果也是相同的。

Java用同样的编程思想，得到的结果也有相似性。
这种错误，主要是初学者对于数据存储原理没有理解清楚，或者，没有意识到内存存储长度的动态变化导致的，下面介绍几种Python删除指定列表索引的方法

a = [3, 4, 5, 6]
b = [1,2]  # b 中为 a的索引

# 删除a中下标为b中的元素
# ====> [3, 6]

# 方法一
a = [a[i] for i in range(len(a)) if (i not in b)]
print(a)  # 重新生成满足条件的list并赋值给 a 


# 方法二
[ a.remove(cElement) for cElement in [a[i] for i in b]]
print(a) # 强制使用 a.remove() 方法方式


# 方法三
a = {a[i] if i not in b else None for i in range(len(a))}
a.remove(None) # 将要删除的对象设置为默认值，
				#  前提是返回结果无序要求以及 纯一种类型
print(a)


# 方法四
m = {i:element for i, element in enumerate(a)}
[m.pop(i) for i in b]
print(list(m.values())) # 将每一个元素对象编码，然后按指定的索引删除，
						 # 并返回 values() 集合对象 --> list对象


# 方法五
class ListSub:

  def __init__(self, list1=None):
    if list1 is None:
      self.listA = []
    elif len(list1):
      self.listA = [x for x in list1]


  def __sub__(self, rhs):
    [self.listA.remove(cElement) for cElement in [self.listA[i] for i in rhs.listA]]
    return self.listA


a = ListSub(a)
b = ListSub(b)

print(a-b)  # 自己写个类，既然Python 不支持 “-” 这个符号，
			 # 而喜欢存储索引的同学，自己写个工具类，实现减法，
			 # 返回自己想要的结果对象。

当然，以上的方式都只是纯粹将方法的实现，没有进行算法优化，以及相应的异常捕获和处理，而Python的效率是有名的…, 所有大牛们请不要过多介怀。

方法其实还有很多，但就项目中出现的Java语法这种错误，我就用Python 来反省一下自己，希望能帮到大家，谢谢。

最后，在学习其他语言的同时，才知道，Python有他语言的便捷之处，也有其他语言共有的缺憾，没有语言基础的使用者，坑尤其多(比如List类的使用)，所以，还要广学习，多对比，写出集多种语言精华的代码，然更多初学者将精力放在业务处理和程序的实现上。

上一篇：详解利用python识别图片中的条码（pyzbar）及条码图片矫正和增强
下一篇：详解Pytorch显存动态分配规律探索