极客教程Python Pandas ŌĆō 查找一列的最大值并返回相应的行值
在 Pandas 中，要查找一列的最大值并返回其相应的行值，可以使用 df.loc [df [col].idxmax()] 。让我们举个例子来更好地理解它。
步骤
- 创建一个二维的、大小可变的、可能不同的表格数据df。
- 打印输入的 DataFrame，df。
- 初始化一个变量 col，以查找该列的最大值。
- 使用df.loc [df[col].idxmax()] 找到最大值及其相应的行。
- 打印第 4 步的输出。
示例
import pandas as pd
df = pd.DataFrame(
{
"x": [5, 2, 7, 0],
"y": [4, 7, 5, 1],
"z": [9, 3, 5, 1]
}
)
print "输入的 DataFrame 如下：\n", df
col = "x"
max_x = df.loc[df[col].idxmax()]
print "列 ", col, " 的最大值及其相应的行值为：\n", max_x
col = "y"
max_x = df.loc[df[col].idxmax()]
print "列 ", col, " 的最大值及其相应的行值为：\n", max_x
col = "z"
max_x = df.loc[df[col].idxmax()]
print "列 ", col, " 的最大值及其相应的行值为：\n", max_x
输出
输入的 DataFrame 如下：
x y z
0 5 4 9
1 2 7 3
2 7 5 5
3 0 1 1
列 x 的最大值及其相应的行值为：
x 7
y 5
z 5
Name: 2, dtype: int64
列 y 的最大值及其相应的行值为：
x 2
y 7
z 3
Name: 1, dtype: int64
列 z 的最大值及其相应的行值为：
x 5
y 4
z 9
Name: 0, dtype: int64