生成多个变量的函数的分组依据

问题描述

我正在尝试使用技术分析库 (TA-lib) 计算一些分组项目: https://mrjbq7.github.io/ta-lib/

某些函数如 AROON 会生成两个变量,AR_UP 和 AR_DOWN。

如果不进行分组,我会使用以下内容:

dft['AR_UP'],dft['AR_DOWN'] = ta.AROON(dft['High'],dft['Low'],14)

它会在 dft 中生成 AR_UP 和 AR_DOWN

但是,当我尝试应用分组依据时:

grouped=dft.groupby(["StockCode"]).apply(lambda x: (ta.AROON(x['High'],x['Low'],14)))

这给了我分组:

StockCode
ABA    ([nan,nan,...
ABP    ([nan,...
ABW    ([nan,...
ACQ    ([nan,...
ACU    ([nan,...
                             ...                        
WTL    ([nan,...
WZR    ([nan,...
XPL    ([nan,...
YBR    ([nan,...
Z1P    ([nan,...
Length: 282,dtype: object

谁能告诉我哪里出错了?

我还想将其重新分配回原始数据帧,例如:

grouped=(grouped.reset_index()
.groupby("StockCode",as_index=False)
.apply(lambda x: x.assign (AR_UP,AR_DOWN=(ta.AROON(x['High'],14))))
.set_index('index') )

有可能吗?

谢谢!

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)