How to append a column to a dataframe with values based on condition

Question

I have the following dataframe:

Country is actually the index:

       2014          2015        PopEst  
Country                                                       
China               8.230121e+12  8.797999e+12  1.367645e+09  
United States       1.615662e+13  1.654857e+13  3.176154e+08  
Japan               5.642884e+12  5.669563e+12  1.274094e+08  
United Kingdom      2.605643e+12  2.666333e+12  6.387097e+07  
Russian Federation  1.678709e+12  1.616149e+12  1.435000e+08  
Canada              1.773486e+12  1.792609e+12  3.523986e+07  
Germany             3.624386e+12  3.685556e+12  8.036970e+07  
India               2.200617e+12  2.367206e+12  1.276731e+09  
France              2.729632e+12  2.761185e+12  6.383735e+07  
South Korea         1.234340e+12  1.266580e+12  4.980543e+07  
Italy               2.033868e+12  2.049316e+12  5.990826e+07  
Spain               1.375605e+12  1.419821e+12  4.644340e+07  
Iran                4.639027e+11           NaN  7.707563e+07  
Australia           1.272520e+12  1.301251e+12  2.331602e+07  
Brazil              2.412231e+12  2.319423e+12  2.059153e+08

And I have the following dict:

ContinentDict  = {'China':'Asia', 
                  'United States':'North America', 
                  'Japan':'Asia', 
                  'United Kingdom':'Europe', 
                  'Russian Federation':'Europe', 
                  'Canada':'North America', 
                  'Germany':'Europe', 
                  'India':'Asia',
                  'France':'Europe', 
                  'South Korea':'Asia', 
                  'Italy':'Europe', 
                  'Spain':'Europe', 
                  'Iran':'Asia',
                  'Australia':'Australia', 
                  'Brazil':'South America'}

I need to append a column showing the Continent Name for each country.

how can I do this?

Maybe best is closing by this

jezrael
– jezrael

2021-02-26 08:02:19 +00:00
Commented Feb 26, 2021 at 8:02 — jezrael
– jezrael, Commented Feb 26, 2021 at 8:02

Mayank Porwal · Accepted Answer · 2021-02-26 07:43:39Z

1

Use:

df['Continent'] = df.index.map('ContinentDict')

answered Feb 26, 2021 at 7:43

Mayank Porwal

34.2k9 gold badges45 silver badges65 bronze badges

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

Comments

Inputvector · Accepted Answer · 2021-02-26 08:07:21Z

0

Try this:

df['Continent'] = df.apply(lambda row : ContinentDict[row.name]  ,axis = 1)

Output:

              2014           2015           PopEst           Continent
China         8.230121e+12  8.797999e+12    1.367645e+0       Asia
United States 1.615662e+13  1.654857e+13    3.176154e+0       North America

answered Feb 26, 2021 at 8:07

Inputvector

1,09310 silver badges22 bronze badges

Collectives™ on Stack Overflow

How to append a column to a dataframe with values based on condition

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related