Python: Finding corresponding indices for an intersection of two lists

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[\'date         


        

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 ]