如何将国际奥委会国家代码转换为国家名称？

Question

我有一个熊猫数据框

import pandas as pd

s = pd.DataFrame({'ioc' : ['ESP', 'CYP', 'USA', 'ESP', 'NED']})

我想回来

out = pd.DataFrame(
    {'ioc' : ['ESP', 'CYP', 'USA', 'ESP', 'NED'],
     'countryName' : ['Spain', 'Cyprus', 'United States of America',
                      'Spain', 'Netherlands']})

Answer 1

import pycountry

def foo(a):
    country = pycountry.countries.get(alpha_3=a)
    return country.name if country else "NA"
    
s.ioc.apply(foo)
0            Spain
1           Cyprus
2    United States
3            Spain
4               NA
Name: ioc, dtype: object

另外一个选择

import country_converter as coco
cc = coco.CountryConverter()
cc.convert(names=s.ioc, to='name')
# ['Spain', 'Cyprus', 'United States', 'Spain', 'not found']

您还可以从country_converter存储库读取数据。

import json
import urllib.request

with urllib.request.urlopen(
    "https://raw.githubusercontent.com/ohitsdaniel/country./master/lib/data/masterData.json"
) as url:
    data = json.loads(url.read().decode())
d = {
    x.get("ioc"): x.get("names")["en"][0]
    for x in data["code"]["ISO2"].values()
    if x.get("ioc")
}
s.ioc.apply(lambda x: d.get(x, "NA"))
0            Spain
1           Cyprus
2    United States
3            Spain
4      Netherlands
Name: ioc, dtype: object