str.contains リストを使って一括検索

Question

初めて質問させていただきます。

前提・実現したいこと

pythonのstr.containsを使う際に
リストを使って一括検索がしたいのですがうまくできません。

dfのA列の中に、listにある単語のどれか一つでも含まれていたら、B列に1を返したいです。

大量の単語を検索したいため、リストでできたらいいなと思い質問させていただきました。
よろしくお願いいたします。

エラーコード

TypeError: first argument must be string or compiled pattern

該当のソースコード

lst1 = ("あかさ","えええ","うえお")
lst2 = ("あ","い","う")
df = pd.DataFrame(lst1)
df.columns = ["A"] 
df['B'] = 0

# こちらは成功
df.loc[df['A'].str.contains('あ'), "B"] = "1"

# 失敗
df.loc[df['A'].str.contains(list), "B"] = "1"

Answer 1

str.contains() は正規表現が使えるので、正規表現 or で 'あ|い|う' を str.contains に指定すればいいと思います。正規表現は "|".join(lst2) で作成すればよいです。

正規表現での、OR（いずれか、または）の表現方法 | WWWクリエイターズ

import pandas as pd

lst1 = ("あかさ", "えええ", "うえお")
lst2 = ("あ", "い", "う")
df = pd.DataFrame({"A": lst1, "B": [0, 0, 0]})

# 失敗
df.loc[df["A"].str.contains("|".join(lst2)), "B"] = "1"
df

Comments

[rukaff]

さっそくの回答ありがとうございます！！
やりたいことが実現でき助かりました！

今後ともどうぞよろしくお願いいたしますm(__)m