Python | 熊猫 DataFrame.idxmax（）介绍

简介

DataFrame.idxmax()函数用于获取数据框(DataFrame)中每个列中最大值对应的索引位置。它返回一个有列标签为索引的Series。

语法

DataFrame.idxmax(axis=0, skipna=True)

参数说明

• axis: 指定轴，0为列，1为行，默认为0
• skipna: 是否跳过NaN值，默认为True

示例

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3, 4, 5], 'B': [5, 4, 3, 2, 1], 'C': [3, 2, 1, 5, 4]})
print(df)

print(df.idxmax())

输出结果：

   A  B  C
0  1  5  3
1  2  4  2
2  3  3  1
3  4  2  5
4  5  1  4

A    4
B    0
C    3
dtype: int64

在这个例子中，我们创建了一个数据框df，使用idxmax()函数获取每列中最大值对应的位置。

使用场景

在数据分析中，我们通常需要得到数据中最大值对应的行或列的位置，这时可以使用idxmax()函数获取对应的索引位置。