大家好
小弟在学习dataframe路上,碰到的案例中
会有需要df然后row by row的执行计算的情况
甚至会需要前一列计算出来的结果
先附上范例
import pandas as pd
df=pd.DataFrame(data=[[1, 7], [3, 6], [5, 4]], columns=['A', 'B'])
def d(row):
if row['A'] > row['B']:
return pd.Series((1, row['A']+row['B']))
else:
return pd.Series((0, row['A']-row['B']))
df[['C', 'D']]=df.apply(d, axis=1)
上述我会得到一个
index | A | B | C | D