时间:2021-01-02 15:11:36 | 栏目:Python代码 | 点击:次
背景:dataFrame的数据,想对某一个列做逻辑处理,生成新的列,或覆盖原有列的值
下面例子中的df均为pandas.DataFrame()的数据
1、增加新列,或更改某列的值
df["列名"]=值
如果值为固定的一个值,则dataFrame中该列所有值均为这个数据
2、处理某列
df["列名"]=df.apply(lambda x:方法名(x,入参2),axis=1)
说明:
1、方法名为单独的方法名,可以处理传入的x数据
2、x为每一行的数据,做为方法的入参1;x中的数据可以用【x.列名】来获取
3、入参2等为方法需要的其他参数,不需要可以不写
4、axis=1,表示每次取一行数据进行处理,按行处理
例子如下:
根据列title的是否包含特定词,来赋值给新的列1或0的值
bugInfo['IntegrationTest'] = bugInfo.apply(lambda x: self.bug_rule(x, "IntegrationTest"), axis = 1) def bug_rule(self, frame, type): # 处理列表中的数据,更新到sql数据库中 result = "0" if type == "SmokeTest": # 冒烟测试 if re.search("^\[冒烟\]|\[冒烟测试\]|【冒烟】|【冒烟测试】", frame["title"]): result = "1" elif type == "InterfaceTest": # 接口测试 if re.search("^\[接口\]|\[接口测试\]|【接口】|【接口测试】", frame["title"]): result = "1" elif type == "IntegrationTest": # 集成测试 if self.IntegrationTime != "" and self.IntegrationTime == frame["created_time"]: result = "1" return result