Python - 迭代数据框列表以创建新变量

Question

我是 Python 新手，想使用 for 循环寻求帮助

我有一个股票代码和价格的数据框，我使用 groupby 函数将其拆分为各自的股票代码，如下所示

grouped = list(dft.groupby(dft.StockCode))

我想要做的是一个 for 循环，它遍历数据帧列表以在每个数据帧中创建一个新变量。例如，我想计算每个股票代码的 RSI。

for df in grouped:
    df['RSI']=ta.RSI(df['Close'],14)

我可以通过拆分每个股票代码然后计算它来单独执行此操作，但这将是一个非常耗时的过程，我想知道是否有更优雅的解决方案。

提前致谢！

Answer 1

这个怎么样？

dft_new = dft.groupby('StockCode').apply(lambda x: ta.RSI(x, 14))

我还没有测试过。但想法是使用 pandas apply on groups。