使用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