pandasでの処理で列同士を比較して同じ値なら値を消す処理をしたいです

Question

前提・実現したいこと

pandasでの処理で列同士を比較して同じ値なら削除するコードを記述したいです。
ご教示よろしくお願いいたします。

該当のソースコード

import pandas as pd

df = pd.DataFrame({"A":["a","b","c","e","d","c"],
                   "B":["c","b","a","e","d","a"]
                  })

このデータフレームでA列とB列が同じ値ならB列の該当箇所をNanにしたいです。
以下の画像のように

試したこと

[None if (x == y for y in df["B"]) else x for x in df["A"]]
これでは、全てNoneになります。
上手い方法がありましたらご教示いただけると幸いです。

Answer 1

whereを使います。

>>> import pandas as pd
>>>
>>> df = pd.DataFrame({"A":["a","b","c","e","d","c"],
...                    "B":["c","b","a","e","d","a"]
...                   })
>>>
>>> df['B'] = df['B'].where(df['A']!=df['B'])
>>> print(df)
   A    B
0  a    c
1  b  NaN
2  c    a
3  e  NaN
4  d  NaN
5  c    a

Comments

[oobutamaru]

無事に出来ました。
お忙しい中回答いただき、有り難うございました。
非常に助かりました。