Pandas DataFrame의 각 행에 대한 최대값이 있는 열 이름을 검색하는 방법은 무엇입니까?

각 행의 최대값 열 이름 검색

다양한 열과 행으로 구성된 DataFrame에서 일반적인 작업은 각 행의 최대값이 있는 열입니다. 다음 DataFrame을 고려하세요.

통신 및 검색 비즈니스 일반 라이프스타일<br>0 0.745763 0.050847 0.118644 0.084746<br>0 0.333333 0.000000 0.583333 0.083333<br>0 0.617021 0.042553 0.297872 0.042553<br>0 0.435897 0.000000 0.410256 0.153846<br>0 0.358974 0.076923 0.410256 0.153846<br>

우리의 목표는 각 행의 최대값과 연관된 열 이름을 포함하는 'Max'라는 라벨이 붙은 새 열을 만드는 것입니다. 원하는 출력은 다음과 유사합니다.

Communications and Search Business General Lifestyle Max<br>0 0.745763 0.050847 0.118644 0.084746 Communications <br>0 0.333333 0.000000 0.583333 0.083333 비즈니스 <br>0 0.617021 0.042553 0.297872 0.042553 커뮤니케이션 <br>0 0.435897 0.000000 0.410256 0.153846 커뮤니케이션 <br>0 0.358974 0.076923 0.410256 0.153846 비즈니스 <br>

이를 달성하기 위해 idxmax 함수를 사용할 수 있습니다.

import pandas as pd

# Create a DataFrame
df = pd.DataFrame({
    'Communications and Search': [0.745763, 0.333333, 0.617021, 0.435897, 0.358974],
    'Business': [0.050847, 0.000000, 0.042553, 0.000000, 0.076923],
    'General': [0.118644, 0.583333, 0.297872, 0.410256, 0.410256],
    'Lifestyle': [0.084746, 0.083333, 0.042553, 0.153846, 0.153846]
})

# Find the column index with the maximum value in each row
max_column_idxs = df.idxmax(axis=1)

# Create a new column with the column names
df['Max'] = max_column_idxs

# Display the updated DataFrame
print(df)

axis 매개변수가 1로 설정된 idxmax 함수를 활용합니다. , 각 행의 최대값으로 열 인덱스를 결정합니다. 그런 다음 이 정보는 각 행의 최대값에 해당하는 열 이름을 식별하는 'Max'라는 새 열을 만드는 데 사용됩니다. 결과 DataFrame은 요청된 형식을 나타냅니다.

위 내용은 Pandas DataFrame의 각 행에 대한 최대값이 있는 열 이름을 검색하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!