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

从熊猫数据帧计算 RSI

1 个答案: