如何使用正則表達式和條件替換熊貓列中的值

Question

我試圖使用正則表達式替換熊貓列（數據框）中的某些值，但是我想基於另一列中的值應用正則表達式。

一個基本的例子；

index  col1  col2
1      yes   foobar
2      yes   foo
3      no    foobar

使用以下內容；

df.loc[df['col1'] == 'yes', 'col2'].replace({r'(fo)o(?!bar)' :r'\1'}, inplace=True, regex=True)

我期望得到以下結果；

index  col1  col2
1      yes   foobar
2      yes   fo
3      no    foobar

但是它似乎不起作用？ 它不會引發任何錯誤或settingwithcopy警告，它什么也不做。 有替代方法嗎？

Answer 1

為了避免鏈接分配，請分配回去並刪除inplace=True ：

mask = df['col1'] == 'yes'
df.loc[mask, 'col2'] = df.loc[mask, 'col2'].replace({r'(fo)o(?!bar)' :r'\1'}, regex=True)

print (df)
  col1    col2
1  yes  foobar
2  yes      fo
3   no  foobar

Answer 2

使用np.where ：

df.assign(
    col2=np.where(df.col1.eq('yes'), df.col2.str.replace(r'(fo)o(?!bar)', r'\1'), df.col2)
)

  col1    col2
1  yes  foobar
2  yes      fo
3   no  foobar