[Python] pandasを使って、特定の列と列の合計値が最大の行を取ってきたいです、、、

Question

Pythonである数字の一覧であるCSVファイルの中身に対して、特定の列と列の合計値が最大になる行を取ってきたいです。具体的には、テストとして以下のようなCSVファイルとコードを考えています。

<具体的にやりたいこと>
列名[A,B,C,D,E,F]の4行×6列のCSVファイルに対して、B列とC列の合計値が最大となる行を取得したいと思っています。

<sample.csv>
[A,B,C,D,E,F]
1,3,7,4,7,9
2,9,6,3,5,1
1,1,1,2,9,5
2,4,2,0,6,1

import pandas as pd

# それぞれの列名を[A,B,C,D,E,F]にする。
df = pd.read_csv('sample.csv', names=['A', 'B', 'C', 'D', 'E', 'F'])
# 「B+C」列の合計値が最大値を取得
x = (df['B + C'].max())
# xが存在する行名を取得
result = df.query('B + C == @x').index[0]

上記のコードを考えていましたが、上手くいきません。
どう書き換えればよいか、わかる方がいらっしゃいましたら教えていただけたら幸いです。よろしくお願いします。

Answer 1

B列とC列の合計値が最大となる行が複数ある場合を考慮すると、以下のようなコードでも実現できます。

import pandas as pd
from io import StringIO

s = """A,B,C,D,E,F
1,3,7,4,7,9
2,9,6,3,5,1
1,1,1,2,9,5
2,4,2,0,6,1
0,9,6,0,0,0"""
df = pd.read_csv(StringIO(s))

bc = df['B'] + df['C']
max = bc.max()

ret = df[ bc == max]
print(ret)
#   A  B  C  D  E  F
#1  2  9  6  3  5  1
#4  0  9  6  0  0  0

Comments

[player31]

回答ありがとうございます！
たしかに最大値が複数ある場合を考えていませんでした。
ベストアンサーに出来なくて、すみません。
とても分かりやすく、参考になりました！

Answer 2

df["B"] + df["C"] で B列、C列を足した Series を作成し、idxmax() で値が最大の要素のインデックスを取得すればよいと思います。

pandas.Series.idxmax — pandas 1.0.0 documentation

import pandas as pd

df = pd.read_csv("sample.csv", names=["A", "B", "C", "D", "E", "F"])

idx = (df["B"] + df["C"]).idxmax()  # 値が最大の行のインデックスを取得
row_with_maxval = df.loc[idx]  # 値が最大の行
print(row_with_maxval)

Comments

[player31]

分かりやすい、回答ありがとうございます！
お陰さまで、やりたいことができました！