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 "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