Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

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