PythonでExcelの「V Look Up」的なことをしたいです。

Question

PythonスクリプトであるCSVファイルの中身に対して、Excelの「V Look Up」的なことをして、特定の値を表示させたいと思っています。
具体的には、以下のようなコードを考えています。

[具体的にやりたいこと]
4行×6列の数字のみのCSVファイルに対して、
それぞれの列名を[A,B,C,D,E,F]と名付け、B列の最大値および、その最大値と同じ行のD列の値を表示させたいと思っています。

[CSVファイルの中身]
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

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

x = (df['B'].max())

result = df.query('B == @x')

y = result['D']

print('B列の最大値は、',x,'で、その時のD列の値は、',y,'です。')

考えた、上記のコードで実行すると以下のような実行結果になってしまいます。

[実行結果]
B列の最大値は、　2　　　9
Name: B, dtype: objectで、その時のD列の値は、　2　　　3
Name: B, dtype: objectです。

[表示させたい結果]
B列の最大値は、9で、その時のD列の値は、3です。

[表示させたい結果]にするために、今のコードを、どのように書き換えればよいのかわかりません。
わかる方がいたら、ぜひ教えて頂きたいです。
また、もっと簡単にできる方法があれば知りたいです。
よろしくお願いします。

Answer 1

元々のコードのままででB列の最大値xは正しく算出されていると思いますので、残りも質問者のコードをそのまま使うと、以下の感じでどうでしょう。

import pandas as pd

df = pd.read_csv('sample.csv', names=['A', 'B', 'C', 'D', 'E', 'F'])
x = (df['B'].max())
# xが存在する行名を取得
result = df.query('B == @x').index[0]
# 行名とD列で値を取得
y = df.at[result,'D']
print('B列の最大値は、',x,'で、その時のD列の値は、',y,'です。')

ただ、これだと最大値が複数行に存在する場合（B列をA列に変更した場合など）に対応できないので、

import pandas as pd

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

# xが存在する行名をリストで取得
results = list(df.query('B == @x').index)

for result in results:
    # 行名とD列で値を特定
    y = df.at[result,'D']
    print('B列の最大値は、',x,'で、その時のD列の値は、',y,'です。')

としてみるとか。B列をA列に変えると、「最大値は…」が2回出力されますが。

Comments

[player31]

教えて頂き、ありがとうございます。
お陰様で、やりたいことができました！

Answer 2

df[df.B == max(df.B)].C.values

#[3]
#<class 'numpy.ndarray'>

※最大値が2つ以上あった場合は、上記の値も2つ以上となります。（例：[4 3]）