matplotlibのステムプロットのベースラインを軸の限界まで延長する

Question

matplotlibでステムプロットを生成するときに最後のデータポイントで終了するのではなく、軸の限界までベースラインを延長するにはどうすればよいですか？

Answer 1

plt.stemプロットのベースラインが

markerline, stemlines, baseline = plt.stem(x, np.cos(x), '-.') 

、関数呼び出しから返されたあなたが[0,1]すべきラインのデータを設定し、yaxis_transformを使用することができ、それらの座標になるように軸座標として解釈される。

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(baseline, 'color', 'r', 'linewidth', 2) 

baseline.set_xdata([0,1]) 
baseline.set_transform(plt.gca().get_yaxis_transform()) 

plt.show()