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

新列生成：

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