检查列表中的一个单词并创建一个新变量

Question

我想使用列表中的城市列表获取 obras desc 内的位置。

我有以下数据框

obras = pd.DataFrame([['1','Agua de Buenos Aires'],['2', 'Sistenas de carreteras Jujuy'],['3','Reasentamiento en Entre Ríos'], ['4','Rutas en Córdoba']],
columns = ['id', 'desc'])

还有名单

list = ['Buenos Aires', 'Jujuy', Corrientes', 'Entre Ríos']

我尝试这样做

for s in obras["desc"]:if any(xs in s for xs in list):obras['Localidad'] = s 

预期的结果是：

id	desc	localidad
1	Agua de Buenos Aires	Buenos Aires
2	Sistenas de carreteras	Jujuy
3	Reasentamiento en Entre Ríos	Entre Ríos
4	Rutas en Córdoba	NaN

但我得到的结果是：

id	desc	localidad
1	Agua de Buenos Aires	Reasentamiento en Entre Ríos
2	Sistenas de carreteras	Reasentamiento en Entre Ríos
3	Reasentamiento en Entre Ríos	Reasentamiento en Entre Ríos
4	Rutas en Córdoba	Reasentamiento en Entre Ríos

我该如何解决这个问题？

谢谢！！！

Answer 1

您可以使用apply检查字符串中是否存在列表项：

import pandas as pd

obras = pd.DataFrame([['1','Agua de Buenos Aires'],['2', 'Sistenas de carreteras Jujuy'],['3','Reasentamiento en Entre Ríos'], ['4','Rutas en Córdoba']],columns = ['id', 'desc'])

list_ = ['Buenos Aires', 'Jujuy', 'Corrientes', 'Entre Ríos']
obras['localidad'] = obras['desc'].apply(lambda x: next(iter([i for i in list_ if i in x]), None))

请注意 - 给定所需的输出 - 这只会在多个匹配的情况下返回第一个匹配。

	id	desc	localidad
0	1	Agua de Buenos Aires	Buenos Aires
1	2	Sistenas de carreteras Jujuy	Jujuy
2	3	Reasentamiento en Entre Ríos	Entre Ríos
3	4	Rutas en Córdoba

PS。不要使用list作为变量名，因为它也是一个内置的python函数。