python - Pandas 替换值

Question

我有以下数据框:

     col
0    pre
1    post
2    a
3    b
4    post
5    pre
6    pre

我想将数据框中不包含“pre”的所有行替换为“nonpre”，因此数据框看起来像:

     col
0    pre
1    nonpre
2    nonpre
3    nonpre
4    nonpre
5    pre
6    pre

我可以使用字典和 pandas 替换来做到这一点，但是我只想选择不是“pre”的元素并将它们替换为“nonpre”。如果不在字典中列出所有可能的 col 值，有没有更好的方法来做到这一点？

Answer 1

只要您熟悉 pandas 允许的 df.loc[condition, column] 语法，这很容易，只需执行 df['col'] != 'pre' 查找应更改的所有行:

df['col2'] = df['col']
df.loc[df['col'] != 'pre', 'col2'] = 'nonpre'

df
Out[7]: 
    col    col2
0   pre     pre
1  post  nonpre
2     a  nonpre
3     b  nonpre
4  post  nonpre
5   pre     pre
6   pre     pre