首页 > 开发 > Python > 正文

pandas中怎样优雅地按行对各列数据进行函数操作?

2017-09-06 19:11:34  来源:网友分享

这里有个很复杂的函数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']))