ifを使って、データフレームの2列の値が同一の時、新しい列に指定したキーワードを返したい

Question

お世話になっております。
お忙しいところ大変恐縮ですが、ご教示頂けますと幸いでございます。

下記のようなデータフレームがあるとして

|名前　|購入ID|初回購入ID|
|ぱんだ| 0001| 0001|
|うさぎ| 0002| 0002|
|ぞう　| 0003| 0003|
|たぬき| 0004| 0004|
|ぱんだ| 0005| 0001|
|わに　| 0006| 0006|

購入IDと初回購入IDが一致するときに”新規”それ以外は”既存”を返したいと考えております。
返す列は新しく作成します。

|名前　|購入ID|初回購入ID|新規既存
|ぱんだ| 0001| 0001|　　新規
|うさぎ| 0002| 0002|　　新規
|ぞう　| 0003| 0003|　　新規
|たぬき| 0004| 0004|　　新規
|ぱんだ| 0005| 0001|　　既存
|わに　| 0006| 0006|　　新規

試したコードは下記となります。
1つ目

if df["初回購入ID"] == df["購入ID"]:
    df["新規既存"] = "新規"
else:
    df["新規既存"] = "既存"

結果↓
ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

２つ目

if df["初回購入ID"] == df["購入ID"]:
    result = "新規"
else:
    result = "既存"

df["新規既存"] = result

結果↓
ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

何か根本的に間違えてる気がします。。
大変恐縮ですが、ご教示頂けますと幸いでございます。

何卒、よろしくお願い致します。

Answer 1

if文の条件式として、True/Falseの列ベクトルを渡しているため、どこを見て判断してよいかわからず、ambiguous（曖昧）というエラーになっています。

ループで1行ずつif文で確認していく、という方法も使えますが、せっかくpandasを使っているので、一気に変換する方法を探りましょう。

今回の場合は、

df['新規既存'] = (df['初回購入ID'] == df['購入ID']).map({True: '既存', False: '新規'})

で、目指す動作になると思います。

Comments

[ForestBamboo]

ありがとうございます！
ご教示頂いた方法で実現できました。
お忙しい中、教えていただきましてありがとうございました。
引き続き、よろしくお願いいたします。