Pandasで値に対応した列を抽出する方法について

Question

前提・実現したいこと

pandasにて下記データフレームがあるとします。

ID,Name,faq,fav1,fav2,fav3,answer
1,山田,fav2,猫,鳥,,
2,佐藤,fav1,米,パン,,
3,田中,fav1,テニス,サッカー,バスケ

answer列に下記条件で値を入れたいのですが、
どのようにすればよいかご教示いただけないでしょうか。

・条件
３列目の値に記載されている列名をanswerに入れる
ただし、もしfav3に値が含まれている場合はfav3を入れないといけない。

イメージとしては下記となります。

ID,Name,faq,fav1,fav2,fav3,answer
1,山田,fav2,猫,鳥,,鳥
2,佐藤,fav1,米,パン,,米
3,田中,fav1,テニス,サッカー,バスケ,バスケ

恐れ入りますがよろしくお願い致します。

Answer 1

おはようございます。

問題文読ませていただきました。

こんな感じでしょうか?ご確認のほど、よろしくお願いいたします。????‍♂️

import pandas as pd

df = pd.read_csv('./sample.csv').fillna({'answer': ''})

for index, row in df.iterrows():
    faq = row['faq']
    if faq == 'fav3':
        df.at[index, 'answer'] = 'fav3'
    else:
        df.at[index, 'answer'] = row[faq]
print(df)

<参考>
・pandasでcsv/tsvファイル読み込み
・Pandasでnan値を削除、穴埋めするfillna、dropnaの使い方

Comments

[kokomi]

おはようございます。

ご回答ありがとうございます！
考えすぎて頭がぐるぐるしてしまいましたが、頂いた内容でスッキリしました。
思ったアウトプットが出ることを確認いたしました。

また、参考状もつけていただき重ねてお礼申し上げます????‍♂️

[退会済みユーザー]

いえいえ、こちらこそご丁寧にありがとうございます。

意外とpandas操作難しいですよね〜。

また困ったことがありましたら、気軽にご相談ください〜。

Answer 2

pandasの基本的な行、列操作は分かっているものとして回答します。

難しく考えず、与えられた条件をif文で素直に実装すれば良いかと思います。

条件は、

1. ３列目の値に記載されている列名をanswerに入れる
2. ただし、もしfav3に値が含まれている場合はfav3を入れないといけない。

となっていますが条件としては2が優先なので、先に2の条件でif文を書き、elifで順に

• faqがfav1だったら、answerにfav1列の内容を代入
• faqがfav2だったら、answerにfav2列の内容を代入

と書けばよいでしょう。

Comments

[kokomi]

ありがとうございます！
Psndasの機能という部分で、変な方向に考えすぎてしまいました????

頂いた回答にてすっきりいたしました。
ありがとうございました！