pandasのwhereの条件式について教えてください

Question

前提・実現したいこと

初歩的な質問で本当にすみませんが、
pandasのwhereの条件式でよくわからないところがあるので教えていただきたいです。

該当のソースコード

import pandas as pd


df = pd.DataFrame({'A': [1,2,3,-1,2,-2,5],'B': [3,3,6,-5,2,2,-1]})

df['A'].where(df['A'] > 0, df['B']) 
#0    1
#1    2
#2    3
#3   -5
#4    2
#5    2
#6    5

この場合組み込み関数のifと同じ感覚で捉えると、列Aの各値において0より大きい値を列Bの値で上書きするように思うのですが、実際の動き方としては、0より小さい値に対して上書きされています。比較演算子＞＜が自分の認識と逆の挙動を取っているので混乱しております。
whereの記法について調べてみたのですが、残念ながらこれといった記述も見られず、ぜひ教えていただければと質問させていただきました。よろしくお願いいたします。

Answer 1

お疲れ様です。
ややこしい回答してしまいました。df.locは符号の向きが逆になってます。

df.locでif文やっても、whereでif文やっても抽出する結果は同じです。
whereの場合は、記載した条件で抽出できなかった行（NaNになる）が、引数に指定したdataframeの行に置き換えてねという使い方になります。

where句のお作法が質問者様の感覚と違っててアレ？ってなっているのかなと思った次第です。

Comments

[gogotowel]

そもそもwhereは抽出を主目的とするもので、該当しないものについて引数の行に置き換えるという形で使うんですね。ifと同じ感覚で使えるのかな？と思ってたのでアレレとなってました。
大変ありがとうございました。

Answer 2

結果の遷移を見ていくと楽かも。リファレンスに使い方も書いてあります

df = pd.DataFrame({'A': [1,2,3,-1,2,-2,5],'B': [3,3,6,-5,2,2,-1]})

print(df['A'].where(df['A'] > 0))

print(df.loc[df['A'] < 0, 'A'])

print(df['A'].where(df['A'] > 0, df['B']) )

Comments

[gogotowel]

ご回答ありがとうございます。
例を試してみました。
df.locで処理した場合→Trueの場合のみ変更
whereで処理した場合→Falseの場合のみ変更
ということになるのでしょうか。

Answer 3

df['A'].where(df['A'] > 0, df['B']) は下記のNaNの部分にdf['B']を代入したものになります。

df['A'].where(df['A'] > 0)
#0    1.0
#1    2.0
#2    3.0
#3    NaN
#4    2.0
#5    NaN
#6    5.0

pandasで条件に応じて値を代入（where, mask）

Comments

[gogotowel]

ご回答ありがとうございます。
> Trueの要素はそのまま、Falseの要素を変更可能
ここの理解が間違ってました。
ifとは考え方が全く違うんですね。