python - Python：为两个列表的交集查找相应的索引

Question

这与我今天不久前提出的一个问题有些相关。我将两个列表的交集如下：

    inter = set(NNSRCfile['datetimenew']).intersection(catdate)

我取交集的两个组件属于两个冗长的列表。是否可以获得相交值的索引？（即原始列表的索引）。

我不太确定从哪里开始。

任何帮助是极大的赞赏！

Answer 1

我会创建一个字典来保存原始索引：

ind_dict = dict((k,i) for i,k in enumerate(NNSRCfile['datetimenew']))

现在，像以前一样构建你的集合：

inter = set(ind_dict).intersection(catdate)

现在，要获取索引列表：

indices = [ ind_dict[x] for x in inter ]