dataframe根据条件对某一列数值进行更新

参考
在对dataframe进行运算的过程中会遇到需要根据其他列的条件来更新当前列的数值这种情况,方式如下:

import pandas as pd
df = pd.Dataframe()

def get_c(index1, index2):
    if index1 == 1 and index2 ==1:
        return 1
    return 0
df["A"] = [1, 1, 1, 0, 0, 0, 1]
df["B"] = [0, 1, 1, 0, 0, 1, 1]
df["C"] = [0, 1, 0, 0, 0, 0, 1]
# 若AB均为1,C才为1,其他情况下C必须为0
df["C"] = df.apply(lambda x: get_c(x.A, x.B), axis=1)
print(df["C"].tolist())
# [0, 1, 1, 0, 0, 0, 1]
# 也可以根据此方式新建一列
df["D"] = df.apply(lambda x: get_c(x.B, x.C), axis=1)

还有一些情况是定位到具体位置,然后对该位置的进行赋值/元素替换

# 根据条件定位所在的行
col_ind = df.loc[df["C"] == 1].index
# 对符合条件的行对应B字段进行替换/赋值
source.loc[col_ind, "B"] = 2

参考:
dataframe定位所在行
dataframe赋值

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值