Python 文字列に特定の単語が記載されている場合Flagをたて、複数回繰り返す

Question

前提・実現したいこと

特定の文字列が入った商品タイトルの一覧があり、その文字列内に指定した特定の単語が含まれている場合は、
隣に列を追加し、1とFlagを立てる（その単語がない場合は0と記載する）コードを作成したいのです。

データは全て英文となり、大文字と小文字も識別せずに処理したいと考えております。

例えば、下記のようなデータがあり、

Product_ID, title
55647, The best Nintendo game Mario
87876, mario cart used condition
98998, Luigi mansion game

例えば、Marioという単語にFlagを立てたい場合、
出力結果が

Product_ID, title, Mario
55647, The best Nintendo game Mario, 1
87876, mario cart used condition, 1
98998, Luigi mansion game, 0

このような結果を出力したいです。
上記はMarioという単語のみでしたが、Flagを立てたい単語が５０以上あります。

該当のソースコード

def Mario(input_str):
  if "Mario" in input_str:
      return 1
  return 0

dataset.loc[:, 'Mario'] = dataset.title.apply(Mario)

試したこと

このコードをそれこそ５０以上作れば、やりたいことは実現できるのですが、
できれば、
（'Mario', 'New', 'Brother') など単語を追加しただけで、全ての処理が終わるようなコードが欲しいと考えております。
また、上記のソースコードだと英語の大文字と小文字をしっかり認識してしまうため、
Marioやmario、MARIOなど考えうる全通りが必要となってしまうため、困っております。

If文などを使うのかと思うのですが、どのようなコードを書けばいいのか想像がつかないため、皆様の助けをいただきたく思います。

Answer 1

pandasのデータフレームを使っているようなので、文字列を含むかどうかのチェックはstr.containsに任せてしまい(大文字・小文字の同一視も可能)、apply(int)で真偽値から1/0の数値に変換できます。

import pandas as pd
import io

txt = """
Product_ID, title
55647, The best Nintendo game Mario
87876, mario cart used condition
98998, Luigi mansion game
"""

df = pd.read_csv(io.StringIO(txt), skipinitialspace=True)
# print(df)

lst = ['Mario', 'Luigi', 'Game']
# print(lst)

for target in lst:
    df[target] = df['title'].str.contains(target, case=False).apply(int)

print(df)

   Product_ID                         title  Mario  Luigi  Game
0       55647  The best Nintendo game Mario      1      0     1
1       87876     mario cart used condition      1      0     0
2       98998            Luigi mansion game      0      1     1

Comments

[sabasoba]

なんといとも簡単に、、、素晴らしいです。問題解決しました。