获取不在索引列表中的列表元素

Question

我有一个索引列表：

idx = [1,4,5]

和感兴趣的列表：

mylist = ['a','b','c','d','e','f']

我想从mylist索引不在idx.

所以结果应该是：

['a','c','d']

我也可以拆分mylist为['a','c','d']and ['b','e','f']，因为无论如何我都会使用它们。

一个 numpy 版本是可以的，尽管我现在实际上只有两个列表。

Answer 1

有了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)