python - Adding a function to a string in a pandas dataframe

Question

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

from countryinfo import CountryInfo

CountryInfo('Lebanon').capital()

Would return 'Beirut'

How can i do this?

Answer 1

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'.