从 Dataframe 中的字符串（列表）中过滤项目

Question

我试图从我的 dataframe 中的所有列表中找到该县。 我的 dataframe 看起来像：

address_df

output：

   address

1   (4070, Halleck Street, Emeryville, Alameda Cou...

2   (Rancho Road, Yuba County, California, 95961, ...

3   (Jasper Lane, Yuba County, California, 95692, ...

4   (Yuba County, California, 95692, United States...

由此，我想得到所有带有“县”字样的元素

例如，output 将是：

county

Alameda County

Yuba County

Yuba County

Yuba County

我首先尝试将字符串转换为列表，但后来我意识到每个索引都不同，我不能简单地使用索引取出县。

帮助将不胜感激！

Answer 1

IIUC 这就是你想要的：

address_df[address_df["address"].str.contains("county", case = False)]

Answer 2

这是一个将 dataframe 转换为列表、提取县并创建新 dataframe 的解决方案：

import pandas as pd

df = pd.DataFrame({'address':["(4070, Halleck Street, Emeryville, Alameda County)","(Rancho Road, Yuba County, California, 95961)",
 "(Jasper Lane, Yuba County, California, 95692)", "(Yuba County, California, 95692, United States)"]})

lst = df.to_numpy().tolist()
counties = []
for address in lst:
    address_parts = address[0].split(",")
    for part in address_parts:
        if 'County' in part:
            c = part.replace('(','').replace(')','').lstrip(' ').split(' ')
            counties.append(c[0])
counties_df = pd.DataFrame(counties, columns= ['counties'])

print(counties_df)

#output:
  counties
0  Alameda
1     Yuba
2     Yuba
3     Yuba