python - 用一组新值替换看起来像列中的元素范围并将其余值设置为 0

Question

对于数据框，我将一列中的一组项目替换为一系列值，如下所示:

df['borough_num'] = df['Borough'].replace(regex=['MANHATTAN', 'BROOKLYN', 'QUEENS', 'STATEN ISLAND','BRONX'], value=[1, 2, 3, 4,5])

我想用值 0 替换之前未提及的“Borough”中所有其余元素的问题我也需要使用正则表达式，因为看起来像数据，例如。 07 BRONX，我需要它也被替换为 5 而不是 0

Answer 1

从你之前的问题，使用replace，关于它为什么有效，你可以检查link

s=df.Borough.replace(dict(zip(l,[1,2,3,4,5])),regex=True)
pd.to_numeric(s,errors = 'coerce').fillna(0).astype(int)
Out[44]: 
0    3
1    5 # notice here still change to 5 
2    1
3    2
4    0
Name: Borough, dtype: int32

---

数据输入

df = pd.DataFrame({
    'Borough': ['QUEENS', 'BRONX 777', 'MANHATTAN', 'BROOKLYN', 'INVALID']})
l = ['MANHATTAN', 'BROOKLYN', 'QUEENS', 'STATEN ISLAND','BRONX']