CODEWITHSUNDEEP
pythonindicesset-intersection
user1620716
user1620716

Reputation: 1533

Python: Finding corresponding indices for an intersection of two lists

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!

Upvotes: 7

Views: 13793

Answers (1)

mgilson
mgilson

Reputation: 309929

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 ]

Upvotes: 15

Related Questions