pandas操作多列计算
记录一下,用pandas操作多列计算,得到结果并形成一列新数据的时候,可以用apply来实现。
df = pd.DataFrame({
'a': [1,2,3,4],
'b': [5,6,7,8],
})
def foo(x, y):
return x * y
df['c'] = df.apply(lambda x: foo(x['a'], x['b']), axis=1)
可对多列数据计算。
记录一下,用pandas操作多列计算,得到结果并形成一列新数据的时候,可以用apply来实现。
df = pd.DataFrame({
'a': [1,2,3,4],
'b': [5,6,7,8],
})
def foo(x, y):
return x * y
df['c'] = df.apply(lambda x: foo(x['a'], x['b']), axis=1)
可对多列数据计算。