1. ホーム
  2. python

[解決済み] 凡例に特定の項目だけを表示する Python Matplotlib

2023-01-15 02:37:54

質問

現在、大量の分類学的データの積み上げ棒グラフをプロットしており、凡例に重要な種のみを表示したいと考えています (~500 のうち ~25 を表示したいです)。これを行うための簡単な方法はありますか?以下は、私が持っているコードです。

labels=['0','20','40','60','80','100','120']
ax1=subj1df.plot(kind='barh', stacked=True,legend=True,cmap='Paired', grid=False)
legend(ncol=2,loc=2, bbox_to_anchor=(1.05, 1), borderaxespad=0.)
label1=['Baseline','8h','24h','48h','96h','120h']
ax1.set_yticklabels(label1, fontdict=None, minor=False)
plt.title('Subject 1 Phyla',fontweight='bold')
plt.savefig('Subject1Phyla.eps', format='eps', dpi=1000)
ax1.set_xticklabels(labels)

編集:1つの凡例だけを表示するためにこれを追加してみましたが、空の凡例を返すだけです。

h, l = ax1.get_legend_handles_labels()
legend(l[4],h[4],ncol=2,loc=2, bbox_to_anchor=(1.05, 1), borderaxespad=0.)

どのように解決するのですか?

私はよく、表示したくない凡例に空のラベルを挿入します。非常に簡単な例を作りましたので、参考にしていただければと思います。あなた自身のデータに合わせてこれを微調整する必要がありますが、必要な要素は揃っているはずです。

import matplotlib.pyplot as plt 
import numpy as np

myY=np.random.randint(20, size=10)
myX=np.arange(0,len(myY))

selected=[5,10,15]

fig = plt.figure()
for X,Y in zip(myX,myY):
    if Y in selected:
        mylabel="label = %s"%(Y); mycolor='blue'
    else:
        mylabel=None; mycolor='red'
    plt.scatter(X,Y,50, color=mycolor, label=mylabel)
plt.legend()
plt.show()

これは以下のようなプロットを作成します。