python - 如何根据列表替换列值？

Question

我有一个这样的列表:

x = ['Las Vegas', 'San Francisco, 'Dallas']

数据框看起来有点像这样:

import pandas as pd
data = [['Las Vegas (Clark County), 25], ['New York', 23], 
        ['Dallas', 27]]
df = pd.DataFrame(data, columns = ['City', 'Value'])

我想将 DF 中的城市值“拉斯维加斯(克拉克县)”替换为“拉斯维加斯”。在我的数据框中有多个具有不同名称的城市，需要更改。我知道我可以做一个正则表达式来去掉括号后面的部分，但我想知道是否有一种更聪明、更通用的方法。

Answer 1

使用Series.str.extract使用正则表达式 OR 通过 | 加入列表的值，然后通过 Series.fillna 将不匹配的值替换为原始值:

df['City'] = df['City'].str.extract(f'({"|".join(x)})', expand=False).fillna(df['City'])
print (df)
        City  Value
0  Las Vegas     25
1   New York     23
2     Dallas     27

另一个想法是使用 Series.str.contains with循环，但如果数据帧很大并且list中有很多值，它应该会很慢:

for val in x:
    df.loc[df['City'].str.contains(val), 'City'] = val