pandas リストの文字列に部分一致する行の抽出方法

Question

実現したいこと

python pandas のデータフレーム　で、ある列から、リストの文字に部分一致する行を抽出したいです。

import pandas as pd
df = pd.DataFrame([['北海道', 'hokkaido'], ['東京', 'tokyo'], ['埼玉', 'saitama'], ['千葉', 'chiba'], ['神奈川', 'kanagawa']], columns=["都道府県", "頭文字"])
pd.DataFrame(df)
print(df)

  都道府県       頭文字
0  北海道  hokkaido
1   東京     tokyo
2   埼玉   saitama
3   千葉     chiba
4  神奈川  kanagawa

lst = ['ho','to']
　上記のリストの文字が含まれる行を抽出したいのですが、下記のようにすると、完全一致になるため部分一致が抽出できません。

df2 = df.query('頭文字 in @lst')
print(df2)

  都道府県       頭文字

下記のように、str.contains ですと、文字は設定できるのですが、リストで設定できないです。
df2 = df.query('頭文字.str.contains("ho")')
print(df2)

  都道府県       頭文字
0  北海道  hokkaido

ある列から、リストの文字に部分一致する行の抽出でよい方法があれば教えていただけないでしょうか？

Answer 1

以下のように正規表現でstr.containsを呼び出してやればよいです。

import pandas as pd

df = pd.DataFrame([['北海道', 'hokkaido'], ['東京', 'tokyo'], ['埼玉', 'saitama'], ['千葉', 'chiba'], ['神奈川', 'kanagawa']], columns=["都道府県", "頭文字"])
pd.DataFrame(df)

lst = ['ho','to']
pat = '|'.join(lst)
df2 = df.query('頭文字.str.contains(@pat, regex=True)')
print(df2)
#  都道府県       頭文字
#0  北海道  hokkaido
#1   東京     tokyo

Comments

[matsutai]

ありがとうございます。使わせていていただきます。