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))

我还没有测试过。但是我们的想法是在组上使用熊猫。