python - 如果 pandas Dataframe 列值与单词匹配，则将其替换为列表

Question

我有一个这样的颜色列表:

color = ['green', 'blue', 'red']

我有一个这样的数据框:

df:
col1        col2
 A        dark green
 B        sea blue
 C          blue
 D       exclusive red
 E          green
 F       pale red

我想将 col2 与 color 列表匹配。如果 col2 的任何单词与 color 列表的元素匹配，则将其替换为列表值。

结果数据框将是

 col1          col2
  A            green
  B            blue
  C            blue
  D            red
  E            green
  F            red

使用 pandas 最有效的方法是什么？

Answer 1

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

print (df)
  col1           col2
0    A     dark green
1    B       sea blue
2    C           blue
3    D  exclusive red
4    E          green
5    F           pale <- not matched value

color=['green','blue','red']

pat = r'({})'.format('|'.join(color))
df['col2'] = df['col2'].str.extract(pat, expand=False).fillna(df['col2'])

print (df)
  col1   col2
0    A  green
1    B   blue
2    C   blue
3    D    red
4    E  green
5    F   pale