从pandas数据帧计算RSI

Question

我有一个pandas数据框，其中数据框的每一列都对应于给定股票的收盘价(IBOVESPA-BRASIL)。我想计算每个数据帧股票(df0)的RSI值，并用这个数据(df1)创建一个新的数据帧。我正在尝试使用pandas-ta库，但我被与收盘价对应的参数卡住了。我该如何解决这个问题呢？

#!pip install yfinance
#!pip install pandas-ta

#Used Packages
import pandas as pd
import pandas_ta as ta
from pandas_datareader import data as pdr
import yfinance as yf
yf.pdr_override()
from datetime import datetime
import warnings
warnings.filterwarnings("ignore")

#Analysis Start and End Date
inicio = '2020-04-01'
fim = '2021-04-27'

#STOCKS
lista = ['LCAM3.SA','MOVI3.SA','CARD3.SA','PRIO3.SA']

#GET DATA YFINANCE
data = pd.DataFrame()
stocks=lista
for stock in stocks:
  try:
    data[stock] = pdr.DataReader(stock, data_source="yahoo", start=inicio, end=fim)["Close"]
  except:
    continue
df0 = data

#GET RSI FROM DF0
df1 = df0.ta.rsi(close=df0, length=14, scalar=None, drift=None, offset=None)```

Answer 1

不要相信你可以一次做所有的列。使用列表理解遍历lista并创建新的数据帧。T被转置以获得正确的方向。

df1 = pd.DataFrame([df0.ta.rsi(close=df0[sym], length=14, scalar=None, drift=None, offset=None) for sym in lista]).T
df1.columns = lista