15

私は使っているmatplotlib

plot()またはにbar()ラベルを追加すると、凡例を簡単に配置できます。しかし、それがまたはの場合はどうなりますcontourf()imshow()

色の範囲を提示できる があることは知っていcolorbar()ますが、満足していません。名前(ラベル)を持つような凡例が必要です。

ここに画像の説明を入力

私が考えることができるのは、マトリックスの各要素にラベルを追加してから、凡例()を試して機能するかどうかを確認することですが、値のように要素にラベルを追加する方法は??

私の場合、生データは次のようになります。

1,2,3,3,4
2,3,4,4,5
1,1,1,2,2

たとえば、1 は「草」、2 は「砂」、3 は「丘」などを表します。imshow() は私の場合は完全に機能しますが、凡例はありません。

私の質問は:

  1. 凡例を自動的に追加できる関数はありますか。たとえば、私の場合、次のようにするだけです: someFunction('grass','sand',...)

  2. ない場合、マトリックスの各値にラベルを追加するにはどうすればよいですか。たとえば、マトリックス 'grass' のすべての 1 にラベルを付け、マトリックス 'sand' のすべての 2 にラベルを付ける...などです。

ありがとうございました!

編集

@dnalow のおかげで、本当にスマートです。ただし、正式な解決策があるかどうかはまだ疑問です。

4

4 に答える 4