时间戳

首页 > 解决方案 > 返回 Python 中就业人数最多的国家名称

python - 返回 Python 中就业人数最多的国家名称

问题描述

我正在使用 NumPy，我面临以下挑战：

这是我的数据：

import numpy as np

# First 20 countries with employment data
countries = np.array([
    'Afghanistan', 'Albania', 'Algeria', 'Angola', 'Argentina',
    'Armenia', 'Australia', 'Austria', 'Azerbaijan', 'Bahamas',
    'Bahrain', 'Bangladesh', 'Barbados', 'Belarus', 'Belgium',
    'Belize', 'Benin', 'Bhutan', 'Bolivia',
    'Bosnia and Herzegovina'
])


# Employment data in 2007 for those 20 countries
employment = np.array([
55.70000076,  51.40000153,  50.5       ,  75.69999695,
58.40000153,  40.09999847,  61.5       ,  57.09999847,
60.90000153,  66.59999847,  60.40000153,  68.09999847,
66.90000153,  53.40000153,  48.59999847,  56.79999924,
71.59999847,  58.40000153,  70.40000153,  41.20000076
])

我必须返回给定就业数据中就业率最高的国家的名称，以及该国的就业情况。

这是我的代码：

def max_employment(countries, employment):
    max_country = ?????     
    max_value = employment.max()   

    return (max_country, max_value)

我因返回就业人数最多的国家的名称而陷入困境

标签： pythonnumpy

解决方案

您正在寻找np.argmax，它将返回最大值的索引：

>>> countries[employment.argmax()]
'Angola'

推荐阅读