pythonのilocメソッドの出力結果について

Question

import pandas as pd
df = pd.DataFrame([[40, "a", True],[20, "b", False],[30, "c", False]])
df.index = ["01", "02", "03"]
df.columns = ["A", "B", "C"]

def judge(arg):
    if arg < 50:
        return "low"
    elif arg < 70:
        return "middle"
    else:
        return "high"

df.loc[:, "C"] = df.iloc[:, 0] * 2
df.loc[:, "B"] = df.iloc[:, 2].apply(judge)
_ = df["C"] > 50
df = df[_]
print(df.iloc[0 , 0], df.iloc[1 ,1])

上記の最後から3行目（_ = df["C"] > 50）と2行目（df = df[_]）でやっていることの意味がわかりません。出力結果を40 lowから40 middleに変えていると思うのですが、なぜlow→middleになるのかご教示ください。

Comments

[meg_]

質問のコードはどこかからのコピペですか？

[hentaiman]

変数の代入をやめた形で書き直してみるなど、試せることはあるので試してみてください
なんかの勉強の過程で出てきたコードならならそれで分かるはず

[akpqky]

python資格試験の勉強をしていて、上記のコードを実装し、変数を変えて解読を試みたりしたのですが、わかりません。

Answer 1

まず _ は、この場合、一種の変数名だと考えてください。普通は _ は、複数返り値の関数の捨てる側の値を受けるために使い、こういう使い方はしません。

_ = df["C"] > 50
はdfの"C"列のみを抽出して >50 という論理演算を各要素に対して行うpandasの演算です。
この時点で _ には

01     True
02    False
03     True

が入ります。（01〜03は行ラベルなので、3行1列の縦ベクトルです）

df = df[]
はdfの各行を、 の論理値に応じて抽出するpandasの演算です。
よって、2行目が削除されdfは

     A       B   C
01  40    high  80
03  30  middle  60

となります。

コードのいろいろな部分でprintを入れて変化を観察してみると、理解しやすいと思います。

Comments

[akpqky]

丁寧にご説明いただき、ありがとうございます。ようやく理解できました。