python - 如何根据不同的 Pandas 列中的值替换字符串

Question

我正在清理数据集，如果 B 列中的值与特定字符串匹配，我需要删除 A 列中的格式错误。

A       B
foo//,  cherry
bar//,  orange
bar//,  cherry
bar     apple

因此，在这种情况下，如果 B 列是“樱桃”，我想将 A 列中的“//,”替换为“,”。最终结果将如下所示。

A       B
foo,    cherry
bar//,  orange
bar,    cherry
bar     apple

非常感谢任何建议

Answer 1

您可以简单地编写一个函数，将一行输入为 series , 检查 cherry条件，用 str.replace 修复字符串并返回行。您可以使用 df.apply在 axis=1 .

def fix(s):
    if s['B']=='cherry':
        s['A']=s['A'].replace('//,',',')
    return s

df.apply(fix, axis=1)

        A       B
0    foo,  cherry
1  bar//,  orange
2    bar,  cherry
3     bar   apple