matplotlib を使用して、ステム プロットを作成し、ステム プロットの色を設定し、プロットの凡例を次のように作成します。
import pyplot as plt
...
plots, legend_names = [], []
for x_var in x_vars:
plots.append(plt.stem(plt.stem(dataframe[y_var], dataframe[x_var])))
markerline, stemlines, baseline = plots[x_var_index]
plt.setp(stemlines, linewidth=2, color=numpy_rand(3,1)) # set stems to random colors
plt.setp(markerline, 'markerfacecolor', 'b') # make points blue
legend_names.append(x_var)
...
plt.legend([plot[0] for plot in plots], legend_names, loc='best')
結果は次のようになります。
凡例の最初の点は (グラフに表示されているように) ポイントの色に対応し、2 番目の点は幹/線の色に対応すると推測しています。ただし、ステムとポイントの両方の色は、最終的にグラフ内のポイントの色に対応します。これを修正する方法はありますか?ありがとう。