python - 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?

question from:https://stackoverflow.com/questions/65856096/adding-a-function-to-a-string-in-a-pandas-dataframe

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