Pandas select based on conditions on multiple columns and a set of values

Question

I have a list of locations and a dataframe like below and I want to select rows from a df with city and country columns where the location of the row (city and country) matches any of the pairs in the dataframe. The list of locations can have a bigger number of pairs which makes typing all conditions less desirable solution.

Locations = [(London,Uk), (Paris, US), (Toronto, Canada)]
  
    | City     | Country  | value |
    | -------- | -------- | ----- |
    | London   | Canada   | 10    |
    | London   | UK       | 200   |
    | Paris    | France   | 300   |
    | Toronto  | Canada   | 40    |
    | Paris    | US       | 100   |

BENY · Accepted Answer · 2021-01-22 02:39:02Z

2

Here is one way reindex after set_index

Locations = [('London','Uk'), ('Paris', 'US'), ('Toronto', 'Canada')]
out = df.set_index(['City','Country']).reindex(Locations).reset_index()
out
Out[83]: 
      City Country value
0   London      Uk   NaN
1    Paris      US   100
2  Toronto  Canada    40

answered Jan 22, 2021 at 2:39

BENY

324k22 gold badges176 silver badges250 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Tarik · Accepted Answer · 2021-01-22 17:11:08Z

1

Change you list of pairs to a DataFrame. Next, do a join between the two DataFrames.

edited Jan 22, 2021 at 17:11

answered Jan 22, 2021 at 2:25

Tarik

11.2k2 gold badges31 silver badges47 bronze badges

Collectives™ on Stack Overflow

Pandas select based on conditions on multiple columns and a set of values

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related