基于列表的 Pandas DataFrame 部分字符串匹配

Question

我有一个DataFrame，如下所示。

import pandas as pd
import numpy as np

df = pd.DataFrame([['One person has died after two motorbikes crashed in the Bay of Plenty.', 'The crash occurred at 3.15pm on Bell Rd in Nukuhou south of Whakatāne police said.', 'Another person suffered minor injuries in the crash.', 'WorkSafe had been advised and the Serious Crash Unit was in attendance police said.',
 'One person has died and another has minor injuries following a serious crash in Nukuhou near Whakatāne in Eastern Bay of Plenty.', 'The crash involved two motorcycles and was reported around 3.15pm.', 'The Serious Crash Unit and WorkSafe are attending the scene.']]).T
df.columns = ['col1']

我想在DataFrame 中查找以下字符串列表是否可用“col1”。

loc_list = ['Bay of Plenty', 'Bell Rd', 'Nukuhou']

以下是我尝试过的。

df['location_mapped_title'] = (df.col1.str
                             .findall('|'.join(loc_list))
                             .str[0])
    col1    location_mapped_title
0   One person has died after two motorbikes crash...   Bay of Plenty
1   The crash occurred at 3.15pm on Bell Rd in Nuk...   Bell Rd
2   Another person suffered minor injuries in the ...   NaN
3   WorkSafe had been advised and the Serious Cras...   NaN
4   One person has died and another has minor inju...   Nukuhou
5   The crash involved two motorcycles and was rep...   NaN
6   The Serious Crash Unit and WorkSafe are attend...   NaN

但它不会打印所有匹配的子字符串。例如，在索引 4 处还有另一个“丰盛湾”。如何找到所有匹配项？

Answer 1

试试findall 或extractall：

df.col1.str.findall(f'({"|".join(loc_list)})', flags=re.IGNORECASE)

---

df.col1.str.extractall(f'({"|".join(loc_list)})', flags=re.IGNORECASE)