当前位置：首页 > news >正文

【Pandas】pandas Series argmax

news 来源：原创 2025/5/14 22:36:45

Pandas2.2 Series

Computations descriptive stats

方法	描述
Series.argsort([axis, kind, order, stable])	用于返回 `Series` 中元素排序后的索引位置的方法
Series.argmin([axis, skipna])	用于返回 `Series` 中最小值索引位置的方法
Series.argmax([axis, skipna])	用于返回 `Series` 中最大值索引位置的方法

pandas.Series.argmax

pandas.Series.argmax 是 Pandas 库中用于返回 Series 中最大值索引位置的方法。它会扫描整个 Series 并返回最大值对应的索引。

参数说明

axis：{0 或 ‘index’}
默认为 0，表示沿索引方向操作。对于 Series 来说，这个参数通常不需要设置。
skipna：布尔值，默认为 True
如果为 True，则忽略缺失值（NaN）。如果为 False，并且 Series 中存在缺失值，则返回 NaN。

示例及结果

示例 1：基本用法

import pandas as pd

# 创建一个示例 Series
s = pd.Series([10, 20, 30, 40, 50])
print("原始 Series:")
print(s)

# 使用 argmax 方法获取最大值的索引位置
max_index = s.argmax()
print("\n最大值的索引位置 (使用 argmax):")
print(max_index)

输出结果：

原始 Series:
0    10
1    20
2    30
3    40
4    50
dtype: int64

最大值的索引位置 (使用 argmax):
4

在这个例子中，argmax 返回了最大值 50 对应的索引位置 4。

示例 2：包含重复值的 Series

# 创建一个包含重复值的 Series
s_with_duplicates = pd.Series([10, 30, 20, 30, 50, 50])
print("原始 Series:")
print(s_with_duplicates)

# 使用 argmax 方法获取最大值的索引位置
max_index_duplicates = s_with_duplicates.argmax()
print("\n最大值的索引位置 (使用 argmax):")
print(max_index_duplicates)

输出结果：

原始 Series:
0    10
1    30
2    20
3    30
4    50
5    50
dtype: int64

最大值的索引位置 (使用 argmax):
4

在这个例子中，argmax 返回了第一个最大值 50 的索引位置 4。注意，即使有多个最大值，argmax 只返回第一个出现的最大值的索引。

示例 3：处理缺失值

# 创建一个包含缺失值的 Series
s_na = pd.Series([10, np.nan, 20, 30, np.nan, 50])
print("原始 Series:")
print(s_na)

# 使用 argmax 方法并忽略缺失值
max_index_skipna = s_na.argmax(skipna=True)
print("\n最大值的索引位置 (使用 argmax 并忽略缺失值):")
print(max_index_skipna)

# 使用 argmax 方法不忽略缺失值
max_index_no_skipna = s_na.argmax(skipna=False)
print("\n最大值的索引位置 (使用 argmax 不忽略缺失值):")
print(max_index_no_skipna)

输出结果：

原始 Series:
0    10.0
1     NaN
2    20.0
3    30.0
4     NaN
5    50.0
dtype: float64

最大值的索引位置 (使用 argmax 并忽略缺失值):
5

最大值的索引位置 (使用 argmax 不忽略缺失值):
-1

在这个例子中，当 skipna=True 时，argmax 忽略了缺失值并返回了最大值 50 的索引位置 5；当 skipna=False 时，由于存在缺失值，返回了 -1。