Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python Pandas – 열의 최대값을 찾고 해당 행 값을 반환합니다.

<시간/>

열의 최대값을 찾고 Pandas에서 해당 행 값을 반환하려면 df.loc[df[col].idxmax()]를 사용할 수 있습니다. . 이해를 돕기 위해 예를 들어보겠습니다.

단계

  • 크기가 가변적이며 잠재적으로 이질적인 2차원 테이블 형식 데이터 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