python3使用另一个列表作为索引删除列表中的多个元素

时间:2017-03-17 07:34:02

标签: list del

import numpy as np
m = np.arange(10).tolist()
n = [2, 6, 4]

我想删除列表m中的第2,第6和第4个元素。

del m[n] 

追踪(最近一次通话):   文件"",第1行,in TypeError:list indices必须是整数或切片,而不是list

我试过了:

ns = np.sort(n)

for i in np.arange(len(ns)):
    m.pop(ns[i] - i)

但它会弹出已删除的元素 有没有优雅的方法来完成这项工作?

1 个答案:

答案 0 :(得分:0)

对于这种简单的情况,您可以使用m = np.delete(m, n)

以下是doc:https://docs.scipy.org/doc/numpy/reference/generated/numpy.delete.html

的链接