Adding a function to a string in a pandas dataframe

Question

I have a data frame that contains a column with countries. I want to convert the country names to capital cities. Example of how the function works:

from countryinfo import CountryInfo
    
CountryInfo('Lebanon').capital()

Would return Beirut

How can i do this?

Cdaman · Accepted Answer · 2021-01-23 05:48:12Z

2

you can use the lambda function and pd.apply() like this:

from countryinfo import CountryInfo
df['Capital'] = df['country'].apply(lambda x : CountryInfo(x).capital()

Here you can put your own df's column name in place of 'Capital' and 'country'.

answered Jan 23, 2021 at 5:48

Cdaman

3201 silver badge12 bronze badges

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

1 Answer 1