【Python】既存の列から特定の条件に該当する列を取得したいです

Question

下記のようなデータフレームがあります。
列「max」の値が、列「0」～「4」のうち、どれと等しいのかを探したいです。

そして、「max」の隣にその等しい（該当する）列名0～4を返したいです。

【今あるデータフレーム　df_resultというDF名】
0 1 2 3 4 max
S1 0.078719 0.006870 -0.029533 0.707470 -0.077999 0.707470
S2 0.250577 -0.025477 -0.132891 0.598110 -0.164581 0.598110
S3 0.675187 -0.095478 0.126588 0.120730 -0.111166 0.675187

↓

【実現したい結果　df_resultに1列（clm）を追加したい】
※右端の列が追加された状態です。
0 1 2 3 4 max clm
S1 0.078719 0.006870 -0.029533 0.707470 -0.077999 0.707470 3
S2 0.250577 -0.025477 -0.132891 0.598110 -0.164581 0.598110 3
S3 0.675187 -0.095478 0.126588 0.120730 -0.111166 0.675187 0

ちなみに、今回は既存の列が0～4ですが、0～10まで変わる場合があります。
その数を個別に指定せずに「clm」を取得できると更に嬉しいです。

どなたかお分かりの方がいらっしゃれば、どうぞよろしくお願いいたします。

Comments

[meg_]

> 列「max」の値が、列「0」～「4」のうち、どれと等しいのかを探したいです。
ない場合はどうするのですか？

[PythonD]

ご質問ありがとう御座います。不慣れなもので低評価をいただいてしまいました…。
maxの値は必ず0-4のいずれかと等しい事が分かっております。
初心者なもので、丸投げのようになってしまっているかもしれませんが、助けていただければ幸いです。
何卒よろしくお願い申し上げます。

Answer 1

PandasのAPIを使用する例です。

import pandas as pd

df_result = pd.DataFrame(
    [
        [0.078719, 0.006870, -0.029533, 0.707470, -0.077999, 0.707470],
        [0.250577, -0.025477, -0.132891, 0.598110, -0.164581, 0.598110],
        [0.675187, -0.095478, 0.126588, 0.120730, -0.111166, 0.675187],
    ],
    index=['S1', 'S2', 'S3'],
    columns=[0, 1, 2, 3, 4, 'max'])

df_result['clm'] = df_result.iloc[:, :-1].eq(df_result['max'], axis='index').idxmax(axis='columns')

print(df_result)

#            0         1         2        3         4       max  clm
# S1  0.078719  0.006870 -0.029533  0.70747 -0.077999  0.707470    3
# S2  0.250577 -0.025477 -0.132891  0.59811 -0.164581  0.598110    3
# S3  0.675187 -0.095478  0.126588  0.12073 -0.111166  0.675187    0

Comments

[PythonD]

ありがとうございます。すごいです！！
先ほどmeguさんに教えていただいた方法を読み解いておりましたが、
こちらも勉強させていただきます！
まだまだ基本的な構文が分かっておらず、そこから学ばなくてはなりません…またこのサイトでご相談させていただくかもしれませんが何卒よろしくお願い致します！

Answer 2

※max列はdf_resultデータフレームの最終列に存在するものとします。

for col in df_result.columns[:-1]:
    if sum(df_result['max'] == df_result[col]):
        df_result.loc[df['max'] == df_result[col], 'clm'] = col

# 	0	1	2	3	4	max	clm
# S1	0.078719	0.006870	-0.029533	0.707470	-0.077999	0.707470	3
# S2	0.250577	-0.025477	-0.132891	0.598110	-0.164581	0.598110	3
# S3	0.675187	-0.095478	0.126588	0.120730	-0.111166	0.675187	0

Comments

[PythonD]

早速ありがとうございます!！
中身の意味を頑張って理解していきたいと思います！