Pandas 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」というラベルの付いた新しい列を作成することです。目的の出力は次のようになります。

通信と検索 ビジネス全般 ライフスタイル 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 Business <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)