Jak utworzyć wykres świecowy za pomocą matplotlib w pythonie

Wykres świecowy to rodzaj wykresu finansowego przedstawiającego zmiany cen papierów wartościowych w czasie.

Poniższy przykład pokazuje, jak utworzyć wykres świecowy przy użyciu biblioteki wizualizacji Matplotlib w języku Python.

Przykład: Tworzenie wykresu świecowego w Pythonie

Załóżmy, że mamy następującą ramkę danych pandy, która pokazuje cenę otwarcia, zamknięcia, najwyższą i najniższą cenę określonych akcji w okresie 8 dni:

 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

Możemy użyć poniższego kodu, aby utworzyć wykres świecowy w celu wizualizacji zmian cen tych akcji w ciągu 8 dni:

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

Każda świeca reprezentuje zmianę ceny papieru wartościowego w danym dniu. Kolor świecy mówi nam, czy cena zakończyła się wyżej (zielony) czy niżej (czerwony) w porównaniu z dniem poprzednim.

Możesz dowolnie zmieniać szerokość świeczników i kolory, aby wykres wyglądał tak, jak chcesz.

Na przykład moglibyśmy sprawić, że świece będą jeszcze cieńsze i użyć różnych kolorów, aby przedstawić dni „wysokie” i „niskie”:

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

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak tworzyć inne popularne wykresy w Pythonie:

Jak utworzyć wiele wykresów Matplotlib na jednej figurze
Jak wykreślić histogram z listy danych w Pythonie
Jak tworzyć wykresy pudełkowe według grup w Pythonie