我有一个索引列表:
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)
关于python - 获取不在索引列表中的列表元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45649106/