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 ]