首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
社区首页 >问答首页 >将用户定义的函数应用于pandas数据帧特定的列,并向数据帧添加新列

将用户定义的函数应用于pandas数据帧特定的列,并向数据帧添加新列
EN

Stack Overflow用户
提问于 2020-07-22 17:14:40
回答 1查看 47关注 0票数 0

我创建了一个名为polar(X_relative , Y_relative , Z_relative)的函数,它接受如下所示的3个参数,结果是新的参数,即(azimuth_angle , tilt_angle)。我想将此函数应用于pandas数据帧,其中函数参数是数据帧中的某些列,并希望将函数(azimuth_angle , tilt_angle)的输出参数作为新函数添加到数据帧中,在数据帧中为每行计算函数。

代码语言:javascript
运行
复制
dataframe columns: X , Y, Z , X_relative , Y_relative , Z_relative
dataframe coulmns I expect after applying the function: X , Y, Z , X_relative , Y_relative , Z_relative , azimuth_angle , tilt_angle
EN

回答 1

Stack Overflow用户

发布于 2020-07-22 17:26:47

polar的返回值封装成一个序列:

代码语言:javascript
运行
复制
return pd.Series([azimuth_angle, tilt_angle]) 

然后使用apply

代码语言:javascript
运行
复制
df[['azimuth_angle', 'tilt_angle']] = df.apply(lambda x: polar(x.X_relative, x.Y_relative, x.Z_relative), axis = 1)
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/63030832

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档