python - 如何通过使用条件查找和替换数据框中的值来更新？

Question

我的目标是根据其他列的条件查找并替换特定列中的值。并根据具体的行进行替换。

举个例子:

import pandas
data = [['red', 'not done'], ['red', 'not done'], ['green', 'not done']]
df = pandas.DataFrame(data, columns = ['color', 'status']) 
print(df)

我们有输出 DataFrame:

   color    status
0    red  not done
1    red  not done
2  green  not done

我的目标是让所有的green 颜色状态都变成done 状态。就像:

   color    status
0    red  not done
1    red  not done
2  green    done

我尝试过的:

df['status'] = df['status'].replace(to_replace = [df['color'] == 'green'], value = 'done')

但它什么也没做。

我也尝试过:df['status'] = df.where(cond = [df['color'] == 'green'] , other = 'done') 但这导致ValueError: Array conditional must be same shape as self 我不明白的错误。

如何正确替换我想要的内容？

Answer 1

进行批量更新的一种简单方法是使用 df.loc

df.loc[df.color == 'green', 'status'] = 'done'

   color    status
0    red  not done
1    red  not done
2  green      done