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

Question

我有一个pandas数据帧

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 repo读取数据。

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

Answer 2

使用国际奥委会国家代码列表

ioc = pd.read_html('https://en.wikipedia.org/wiki/List_of_IOC_country_codes')[0]
ioc = ioc.assign(Code=ioc['Code'].str[-3:]).set_index('Code')['National Olympic Committee']
s['countryName'] = s['ioc'].map(ioc)
print(s)
# Output:
ioc    countryName
0  ESP          Spain
1  CYP         Cyprus
2  USA  United States
3  ESP          Spain
4  NED    Netherlands