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

Question

This is somewhat related to a question I asked not too long ago today. I am taking the intersection of two lists as follows:

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

The two components that I am taking the intersection of belong to two lengthy lists. Is it possible to get the indices of the intersected values? (The indices of the original lists that is).

I'm not quite sure where to start with this one.

Any help is greatly appreciated!

Answer 1

I would create a dictionary to hold the original indices:

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

Now, build your sets as before:

inter = set(ind_dict).intersection(catdate)

Now, to get a list of indices:

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