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

问题描述

问 题

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

这篇关于python - pandas中怎样优雅地按行对各列数据进行函数操作？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！