这里有个很复杂的函数foo(),需要根据dataframe的一行中的两列元素进行函数运算,这里怎样写比较优雅?
用for循环也能做,但是很辛苦。
import pandas as pd
def foo(a,b):
'''这里是一个极其复杂的函数'''
return(a*b)
a=pd.DataFrame([1,2,3,4,5])
b=pd.DataFrame([2,3,4,5,6])
c=pd.concat([a,b],axis=1)
c.columns=["a","b"]
c=c[["a","b"]].apply(foo,axis=1)
解决方案
用python的函数式编程
list(map(foo,c['a'],c['b']))