Comment créer un graphique en chandeliers à l’aide de Matplotlib en Python

Un graphique en chandeliers est un type de graphique financier qui affiche les mouvements de prix des titres au fil du temps.

L’exemple suivant montre comment créer un graphique en chandeliers à l’aide de la bibliothèque de visualisation Matplotlib en Python.

Exemple : création d’un graphique en chandeliers en Python

Supposons que nous ayons le DataFrame pandas suivant qui montre le prix d’ouverture, de clôture, haut et bas d’une certaine action sur une période de 8 jours :

import pandas as pd

#create DataFrame
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

Nous pouvons utiliser le code suivant pour créer un graphique en chandelier afin de visualiser les mouvements de prix de cette action au cours de cette période de 8 jours :

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

Chaque chandelier représente l’évolution du prix du titre un jour donné. La couleur du chandelier nous indique si le prix a clôturé à la hausse (vert) ou à la baisse (rouge) par rapport à la veille.

N’hésitez pas à modifier la largeur des chandeliers et les couleurs utilisées pour faire apparaître le graphique comme vous le souhaitez.

Par exemple, nous pourrions rendre les bougies encore plus fines et utiliser différentes couleurs pour représenter les jours « hauts » et « bas » :

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

Ressources additionnelles

Les didacticiels suivants expliquent comment créer d’autres graphiques courants en Python :

Comment créer plusieurs tracés Matplotlib sur une seule figure
Comment tracer un histogramme à partir d’une liste de données en Python
Comment créer des boîtes à moustaches par groupe en Python