Como converter um array de dados nomeado OHLCV em um dataframe numpy?

Dec 09 2020

Meus dados consistem em um objeto OHLCV particular que é um pouco estranho porque só pode ser acessado pelo nome, como este:

# rA = [<MtApi.MqlRates object at 0x000000A37A32B308>,...]
type(rA)
# <class 'list'>

ccnt = len(rA)              # 100
for i in range(ccnt):
    print('{} {} {} {} {} {} {}'.format(i, rA[i].MtTime, rA[i].Open, rA[i].High, rA[i].Low, rA[i].Close, rA[i].TickVolume))

#0 1607507400 0.90654 0.90656 0.90654 0.90656 7
#1 1607507340 0.90654 0.9066  0.90653 0.90653 20
#2 1607507280 0.90665 0.90665 0.90643 0.90653 37
#3 1607507220 0.90679 0.90679 0.90666 0.90666 22
#4 1607507160 0.90699 0.90699 0.90678 0.90678 29

com alguma formatação adicional que tenho:

Time         Open     High     Low      Close     Volume
-----------------------------------------------------------------
1607507400   0.90654  0.90656  0.90654  0.90656   7
1607507340   0.90654  0.90660  0.90653  0.90653   20
1607507280   0.90665  0.90665  0.90643  0.90653   37
1607507220   0.90679  0.90679  0.90666  0.90666   22

Eu tentei coisas assim:

df = pd.DataFrame(data = rA, index = range(100), columns = ['MtTime', 'Open', 'High','Low', 'Close', 'TickVolume'])

# Resulting in:
# TypeError: iteration over non-sequence

Como posso converter essa coisa em um Panda trama de dados, para que eu possa traçar isso usando os nomes originais?


A plotagem usando matplotlib deve ser possível com algo assim:


import matplotlib.pyplot as plt
import pandas as pd
from pandas.plotting import register_matplotlib_converters
register_matplotlib_converters()
...

df = pd.DataFrame(rA)   # not working

df['time'] = pd.to_datetime(df['MtTime'], unit='s')
plt.plot(df['MtTime'], df['Open'], 'r-', label='Open')
plt.plot(df['MtTime'], df['Close'], 'b-', label='Close')
plt.legend(loc='upper left')
plt.title('EURAUD candles')
plt.show()

Possivelmente questões relacionadas (mas não foram úteis para mim):

  • Numpy / Matplotlib - Transforma os dados do tick em OHLCV
  • O agregador OHLC não funciona com dataframe em pandas?
  • Como converter um dataframe do pandas em uma matriz numpy com os nomes das colunas
  • Convertendo Numpy Structured Array em Pandas Dataframes
  • Agregação Pandas OHLC em dados OHLC
  • Abertura, alta, baixa, fechamento por 5 minutos python de dados de estoque
  • Converter dados de estoque OHLC em um período de tempo diferente com python e pandas

Respostas

1 jezrael Dec 09 2020 at 16:02

Uma ideia é usar a compreensão de lista para extrair valores para lista de tuplas:

L = [(rA[i].MtTime, rA[i].Open, rA[i].High, rA[i].Low, rA[i].Close, rA[i].TickVolume) 
      for i in range(len(rA))]

df = pd.DataFrame(L, columns = ['MtTime', 'Open', 'High','Low', 'Close', 'TickVolume']))

Ou se possível:

df = pd.DataFrame({'MtTime':list(rA.MtTime), 'Open':list(rA.Open), 
                   'High':list(rA.High),'Low':list(rA.Low), 
                   'Close':list(rA.Close), 'TickVolume':list(rA.TickVolume)})