Comment faire des lignes de tige plus épaisses dans matplolib

Question

Je veux faire des lignes de tige plus épaisses en python en utilisant plt.stem.

Voici mon code

import matplotlib.pyplot as plt 
import numpy as np 

N = 20 

n = np.arange(0, 2*N, 1) 

x = np.exp(-n/N)*np.exp(1j * 2*np.pi/N*n) 

plt.stem(n,x.real) 

plt.show() 

J'ai changé plt.stem(n,x.real,linewidth=10), mais rien n'a changé. N'y a-t-il aucune fonction pour définir le linewidth dans plt.stem?

Answer 1

La documentation de plt.stem montre que la fonction renvoie tous les objets de ligne créés par le tracé. Vous pouvez l'utiliser pour effectuer manuellement les lignes plus épaisses après avoir tracé:

import matplotlib.pyplot as plt 
import numpy as np 

N = 20 
n = np.arange(0, 2*N, 1) 
x = np.exp(-n/N)*np.exp(1j * 2*np.pi/N*n) 

markers,stems,base = plt.stem(n,x.real) 
for stem in stems: 
    stem.set_linewidth(10) 
plt.show() 

Answer 2

Cela peut également être modifié en utilisant plt.setp() comme il est indiqué dans la documentation matplotlib example. Le plt.setp() method vous permet de définir les propriétés d'un objet artiste après sa création.

import matplotlib.pyplot as plt 
import numpy as np 

x = np.linspace(0.1, 2*np.pi, 10) 
markerline, stemlines, baseline = plt.stem(x, np.cos(x), '-.') 
plt.setp(stemlines, 'linewidth', 4) 

plt.show()