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 ]