2

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 番目の点は幹/線の色に対応すると推測しています。ただし、ステムとポイントの両方の色は、最終的にグラフ内のポイントの色に対応します。これを修正する方法はありますか?ありがとう。

4

1 に答える 1