提取人口最接近的5个国家

Question

我有一个不同国家2013年人口类型序列的对象，其中国家是指数

输入示例：

Country Name     Population in 2013
Aruba            103159.0
Afghanistan      32269589.0
Angola           26015780.0
...              ...

现在我想随机挑选一个国家及其人口。我是这样做的

countr = set(country.name for country in pycountry.countries)
listofcountr=list(countr)
randcountry=random.choice(listofcountr)

现在我想找到5个国家，它们的人口和我随机找到的这个国家的人口最接近。在绝对值的意义上最接近。我怎样才能做到这一点

Answer 1

使用pandas的另一种方法可以是以下方法（请注意，这些值是伪值）——

df = pd.DataFrame({'pop':[10,20,30,15,34,23,10,12], 'country':['a','b','c','d','e','f','g','h']})
df = df.set_index('country')
df
         pop
country
a         10
b         20
c         30
d         15
e         34
f         23
g         10
h         12

现在，如果您想找到值为pop的5个最接近的国家，例如，country b，您可以尝试以下方法-

df['diff'] = (df['pop'] - df.loc['b', 'pop']).abs()
df[df.index != 'b'].sort_values(['diff']).head(5).index.tolist()
['f', 'd', 'h', 'a', 'c', 'g']