Pandasで複数列を比較して新しい列を作成

Question

PandasのデータフレームA列、B列、C列の値を比較してA列が大きければ1、B列なら2、C列なら3という値のD列を作成したいです。
調べても2つの列を比較するサイトしかありませんでした。
ご教授お願い致します。

Comments

[ppaul]

A列がゼロでB列とC列がどちらも10の時はどうしたいですか？

Answer 1

最大値が複数ある場合は一番左にあるものを最大とみなすということで良ければ、以下でできます。

df['D'] = df.set_axis([1, 2, 3], axis=1).idxmax(axis=1)

実行例

>>> import pandas as pd
>>> import random
>>> df = pd.DataFrame([[random.randint(0,10) for i in range(3)] for j in range(5)], columns=list('ABC'))
>>>
>>> print(df)
   A  B  C
0  9  4  6
1  8  6  2
2  2  8  2
3  8  1  8
4  4  1  6
>>> df['D'] = df.set_axis([1, 2, 3], axis=1).idxmax(axis=1)
>>> print(df)
   A  B  C  D
0  9  4  6  1
1  8  6  2  1
2  2  8  2  2
3  8  1  8  1
4  4  1  6  3

参考
pandasで最大値・最小値の行名・列名を取得するidxmax, idxmin

Answer 2

import pandas as pd
import io

data = '''
A,B,C
1,2,3
6,1,5
2,9,1
'''

df = pd.read_csv(io.StringIO(data))

#
df['D'] = df[['A', 'B', 'C']].values.argmax(axis=1) + 1
print(df)

#
   A  B  C  D
0  1  2  3  3
1  6  1  5  1
2  2  9  1  2