pandas Series 的 argmax 方法和 idxmax 方法用于获取 Series 的最大值的索引值:
举个栗子:
有一个pandas Series,它的索引是国家名,数据是就业率,要找出就业率最高的国家:
import pandas as pd
原文链接:https://blog.csdn.net/weixin_34227447/article/details/86130861
获取pandas中series最大值对应的索引。
pandas Series 的 argmax 方法和 idxmax 方法用于获取 Series 的最大值的索引值:
举个栗子:
有一个pandas Series,它的索引是国家名,数据是就业率,要找出就业率最高的国家:
import pandas as pd
countries = [
‘Afghanistan’, ‘Albania’, ‘Algeria’, ‘Angola’,
‘Argentina’, ‘Armenia’, ‘Australia’, ‘Austria’,
‘Azerbaijan’, ‘Bahamas’, ‘Bahrain’, ‘Bangladesh’,
‘Barbados’, ‘Belarus’, ‘Belgium’, ‘Belize’,
‘Benin’, ‘Bhutan’, ‘Bolivia’, ‘Bosnia and Herzegovina’,
]
employment_values = [
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,
]
# Employment data in 2007 for 20 countries
employment = pd.Series(employment_values, index=countries)
可以这样做:
max_country = employment.idxmax()
max_country = employment.argxmax()
# 结果: ‘Angola’
如果是一个没有索引值的Series,则返回它的位置索引:
pure_employment = pd.Series(employment_values) print(pure_employment.argmax()) print(pure_employment.idxmax())
# 结果: 3