เพื่อหาค่าสูงสุดของคอลัมน์และคืนค่าแถวที่สอดคล้องใน 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("Input DataFrame is:\n", df)
col = "x"
max_x = df.loc[df[col].idxmax()]
print("Maximum value of column ", col, " and its corresponding row values:\n", max_x)
col = "y"
max_x = df.loc[df[col].idxmax()]
print("Maximum value of column ", col, " and its corresponding row values:\n", max_x)
col = "z"
max_x = df.loc[df[col].idxmax()]
print("Maximum value of column ", col, " and its corresponding row values:\n", max_x)
ผลลัพธ์
Input DataFrame is: x y z 0 5 4 9 1 2 7 3 2 7 5 5 3 0 1 1 Maximum value of column x and its corresponding row values: x 7 y 5 z 5 Name: 2, dtype: int64 Maximum value of column y and its corresponding row values: x 2 y 7 z 3 Name: 1, dtype: int64 Maximum value of column z and its corresponding row values: x 5 y 4 z 9 Name: 0, dtype: int64