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)
但它会弹出已删除的元素 有没有优雅的方法来完成这项工作?
答案 0 :(得分:0)
对于这种简单的情况,您可以使用m = np.delete(m, n)
。
以下是doc:https://docs.scipy.org/doc/numpy/reference/generated/numpy.delete.html
的链接