pandas 如果满足条件，将值从一列复制到另一列

Question

我有一个数据框：

df = 
col1  col2  col3 
1      2     3
1      4     6
3      7     2

我想编辑 df，当 col1 的值小于 2 时，从 col3.

中取值

所以我会得到：

new_df = 
col1  col2  col3 
3      2     3
6      4     6
3      7     2

我尝试使用 assign 和 df.loc 但没有成功。

最好的方法是什么？

Answer 1

df['col1'] = df.apply(lambda x: x['col3'] if x['col1'] < x['col2'] else x['col1'], axis=1)

Answer 2

您可以考虑使用 apply 函数。

df['col1'] = df.apply(lambda c: c['col3'] if c['col1'] < 2 else c['col1'], axis=1)

编辑：抱歉，我从你的模拟结果中看到你指的是 col2 而不是 2 的 int。Eric Yang 的回答将解决你的问题。

Answer 3

最有效的方法是使用 loc 运算符：

mask = df["col1"] < df["col2"]
df.loc[mask, "col1"] = df.loc[mask, "col3"]

Answer 4

df.loc[df["col1"] < 2, "col1"] = df["col3"]

Answer 5

如@anky_91所述，使用np.where更新'col1'值：

df['col1'] = np.where(df['col1'] < df['col2'], df['col3'], df['col1'])

pandas copy value from one column to another if condition is met