python - 根据列表替换值，但将替换值保留在单独的列中

Question

我有一个像下面这样的 df

ColumnA
"ABC X1"
"BCS X2"
"CCC X3"

如果字符串在列表中，我想用“”替换部分字符串，但也将它们保留在单独的列中。比方说我的 list 是；

["X1","X3"]

所以我想要的输出是

ColumnA     ColumnB 
"ABC"       "X1"
"BCS X2"    
"CCC"       "X3"

我知道如何使用 replace 替换 values 值，我还认为我可以通过使用 for 循环并将每个值与列表中的值进行比较并在它们匹配时保留它们来获得解决方案, 但也许有更好的解决方案？

Answer 1

你可以做两次:

df['ColumnB'] = df.ColumnA.str.extract(f'({"|".join(strs)})')[0]
df['ColumnA'] =df.ColumnA.replace(strs,'', regex=True)

输出:

  ColumnA ColumnB
0    ABC       X1
1  BCS X2     NaN
2    CCC       X3