我有一个索引列表:
idx = [1,4,5]
和感兴趣的列表:
mylist = ['a','b','c','d','e','f']
我想从 mylist 中取出索引不在 idx 中的所有元素。
所以结果应该是:
['a','c','d']
我也可以将 mylist 拆分为 ['a','c','d'] 和 ['b','e ','f']，因为无论如何我都会用到它们。
numpy 版本没问题，但实际上我现在只有两个列表。
请您参考如下方法:
借助 numpy，您可以使用掩码数组。
import numpy as np
x=np.array(mylist)
mask=np.full(len(mylist),True,dtype=bool)
mask[idx]=False
y=x[mask]
z=x[~mask]
print(y,z)
