dataframe,根据多列信息生成新列

使用apply函数

DataFrame创建

import pandas as pd
data = {'year':['2018','2018','2018','2019','2019'],
       'math':[23, 99, 90, 30, 10], 
       'compute': [78, 62, 90, 66, 88]}
df = pd.DataFrame(data)
df

dataframe,根据多列信息生成新列_第1张图片
新列生成:

def total(math, compute):
    if math + compute < 100:
        return 100
    return math + compute

df["sum"] = df.apply(lambda row: total(row.math, row.compute), axis=1)
df

dataframe,根据多列信息生成新列_第2张图片

你可能感兴趣的:(pandas,pandas,python,机器学习)