Como criar um gráfico de velas usando matplotlib em python

Um gráfico de velas é um tipo de gráfico financeiro que exibe os movimentos de preços dos títulos ao longo do tempo.

O exemplo a seguir mostra como criar um gráfico de velas usando a biblioteca de visualização Matplotlib em Python.

Exemplo: Criando um gráfico de velas em Python

Digamos que temos o seguinte DataFrame do pandas que mostra o preço de abertura, fechamento, máximo e mínimo de uma determinada ação durante um período de 8 dias:

 import pandas as pd

#createDataFrame
prices = pd. DataFrame ({' open ': [25, 22, 21, 19, 23, 21, 25, 29],
                       ' close ': [24, 20, 17, 23, 22, 25, 29, 31],
                       ' high ': [28, 27, 29, 25, 24, 26, 31, 37],
                       ' low ': [22, 16, 14, 17, 19, 18, 22, 26]},
                       index=pd. date_range (" 2021-01-01 ", periods=8, freq=" d "))

#display DataFrame
print (prices)

            open close high low
2021-01-01 25 24 28 22
2021-01-02 22 20 27 16
2021-01-03 21 17 29 14
2021-01-04 19 23 25 17
2021-01-05 23 22 24 19
2021-01-06 21 25 26 18
2021-01-07 25 29 31 22
2021-01-08 29 31 37 26

Podemos usar o código a seguir para criar um gráfico de velas para visualizar os movimentos de preços desta ação durante este período de 8 dias:

 import matplotlib. pyplot as plt

#create figure
plt. figure ()

#define width of candlestick elements
width = .4
width2 = .05

#define up and down prices
up = prices[prices. close >=prices. open ]
down = prices[prices. close <prices. open ]

#define colors to use
col1 = ' green '
col2 = ' red '

#plot up prices
plt. bar (up. index , up. close -up. open , width, bottom=up. open , color=col1)
plt. bar (up. index ,up. high -up. close ,width2,bottom=up. close ,color=col1)
plt. bar (up. index , up. low -up. open , width2, bottom=up. open , color=col1)

#plot down prices
plt. bar (down. index , down. close -down. open , width, bottom=down. open , color=col2)
plt. bar (down. index , down. high -down. open , width2, bottom=down. open , color=col2)
plt. bar (down. index ,down. low -down. close ,width2,bottom=down. close ,color=col2)

#rotate x-axis tick labels
plt. xticks (rotation= 45 , ha=' right ')

#display candlestick chart
plt. show () 

Cada vela representa a variação do preço do título em um determinado dia. A cor da vela nos diz se o preço fechou em alta (verde) ou em baixa (vermelho) em relação ao dia anterior.

Sinta-se à vontade para alterar a largura dos castiçais e as cores usadas para fazer o gráfico ter a aparência que você deseja.

Por exemplo, poderíamos tornar as velas ainda mais finas e usar cores diferentes para representar os dias “altos” e “baixos”:

 import matplotlib. pyplot as plt

#create figure
plt. figure ()

#define width of candlestick elements
width = .2
width2 = .02

#define up and down prices
up = prices[prices. close >=prices. open ]
down = prices[prices. close <prices. open ]

#define colors to use
col1 = ' black '
col2 = ' steelblue '

#plot up prices
plt. bar (up. index , up. close -up. open , width, bottom=up. open , color=col1)
plt. bar (up. index ,up. high -up. close ,width2,bottom=up. close ,color=col1)
plt. bar (up. index , up. low -up. open , width2, bottom=up. open , color=col1)

#plot down prices
plt. bar (down. index , down. close -down. open , width, bottom=down. open , color=col2)
plt. bar (down. index , down. high -down. open , width2, bottom=down. open , color=col2)
plt. bar (down. index ,down. low -down. close ,width2,bottom=down. close ,color=col2)

#rotate x-axis tick labels
plt. xticks (rotation= 45 , ha=' right ')

#display candlestick chart
plt. show () 

Recursos adicionais

Os tutoriais a seguir explicam como criar outros gráficos comuns em Python:

Como criar vários gráficos Matplotlib em uma única figura
Como traçar um histograma a partir de uma lista de dados em Python
Como criar box plots por grupo em Python