twinx()を使用した第2軸:凡例に追加する方法は?


288

を使用して、2つのy軸を持つプロットがありますtwinx()。また、線にラベルを付けてで表示したいのですlegend()が、凡例の1つの軸のラベルしか取得できません。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('mathtext', default='regular')

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(time, Swdown, '-', label = 'Swdown')
ax.plot(time, Rn, '-', label = 'Rn')
ax2 = ax.twinx()
ax2.plot(time, temp, '-r', label = 'temp')
ax.legend(loc=0)
ax.grid()
ax.set_xlabel("Time (h)")
ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")
ax2.set_ylabel(r"Temperature ($^\circ$C)")
ax2.set_ylim(0, 35)
ax.set_ylim(-20,100)
plt.show()

したがって、凡例の最初の軸のラベルのみを取得し、2番目の軸の「temp」ラベルは取得しません。この3番目のラベルを凡例に追加するにはどうすればよいですか?

ここに画像の説明を入力してください


4
[ 製品コードの近くのどこでもこれを行わないでください ]私の唯一の目的が適切な凡例を使って美しいプロットをできるだけ早く生成することである場合、私は使用しaxているスタイルで空の配列をプロットする醜いハックを使用しax2ます:であなたの場合、ax.plot([], [], '-r', label = 'temp')。それは適切にそれを行うよりもはるかに高速でシンプルです...
ネインシュタイン

回答:


370

次の行を追加すると、2番目の凡例を簡単に追加できます。

ax2.legend(loc=0)

あなたはこれを得るでしょう:

ここに画像の説明を入力してください

しかし、1つの凡例にすべてのラベルが必要な場合は、次のようにする必要があります。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('mathtext', default='regular')

time = np.arange(10)
temp = np.random.random(10)*30
Swdown = np.random.random(10)*100-10
Rn = np.random.random(10)*100-10

fig = plt.figure()
ax = fig.add_subplot(111)

lns1 = ax.plot(time, Swdown, '-', label = 'Swdown')
lns2 = ax.plot(time, Rn, '-', label = 'Rn')
ax2 = ax.twinx()
lns3 = ax2.plot(time, temp, '-r', label = 'temp')

# added these three lines
lns = lns1+lns2+lns3
labs = [l.get_label() for l in lns]
ax.legend(lns, labs, loc=0)

ax.grid()
ax.set_xlabel("Time (h)")
ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")
ax2.set_ylabel(r"Temperature ($^\circ$C)")
ax2.set_ylim(0, 35)
ax.set_ylim(-20,100)
plt.show()

これはあなたにこれを与えるでしょう:

ここに画像の説明を入力してください


2
これはerrorbarプロットで失敗します。それらを正しく処理するソリューションについては、以下を参照してください。stackoverflow.com
Davide

1
2つの.legend(loc = 0)を指定した場合のように、2つの重複する凡例を回避するには、凡例の位置の値に2つの異なる値(両方とも0以外)を指定する必要があります。参照:matplotlib.org/api/legend_api.html
Roalt

複数のラインがあるサブプロットに1つのラインを追加するときに問題が発生しましたax1。この場合はlns1=ax1.lines、使用してlns2からこのリストに追加してください。
リトルボビーテーブル

が使用するさまざまな値locについて、ここで
Dror

1
より自動的な方法については、以下の回答を参照してください(matplotlib> = 2.1を使用):stackoverflow.com/a/47370214/653364
joris

183

この機能が新しいかどうかはわかりませんが、自分で行とラベルを追跡するのではなく、get_legend_handles_labels()メソッドを使用することもできます。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('mathtext', default='regular')

pi = np.pi

# fake data
time = np.linspace (0, 25, 50)
temp = 50 / np.sqrt (2 * pi * 3**2) \
        * np.exp (-((time - 13)**2 / (3**2))**2) + 15
Swdown = 400 / np.sqrt (2 * pi * 3**2) * np.exp (-((time - 13)**2 / (3**2))**2)
Rn = Swdown - 10

fig = plt.figure()
ax = fig.add_subplot(111)

ax.plot(time, Swdown, '-', label = 'Swdown')
ax.plot(time, Rn, '-', label = 'Rn')
ax2 = ax.twinx()
ax2.plot(time, temp, '-r', label = 'temp')

# ask matplotlib for the plotted objects and their labels
lines, labels = ax.get_legend_handles_labels()
lines2, labels2 = ax2.get_legend_handles_labels()
ax2.legend(lines + lines2, labels + labels2, loc=0)

ax.grid()
ax.set_xlabel("Time (h)")
ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")
ax2.set_ylabel(r"Temperature ($^\circ$C)")
ax2.set_ylim(0, 35)
ax.set_ylim(-20,100)
plt.show()

1
これは、プロットが凡例と重なる軸を処理できる唯一のソリューションです(最後の軸は凡例をプロットする必要があるものです)
Amelio Vazquez-Reina

5
このソリューションはerrorbarプロットでも機能しますが、受け入れられたものは失敗します(線とそのエラーバーは別々に表示され、適切なラベルが付いたものは表示されません)。さらに、それはより簡単です。
Davide

ちょっとした問題:ラベルを上書きしたい場合ax2は機能せず、最初からラベルが設定されていません
CiprianTomoiagăMay

備考:クラシックプロットの場合、ラベル引数を指定する必要はありません。しかし、他の人、例えば。あなたがする必要があるバー。
belka

これにより、何本の線がプロットされるかが事前にわからない場合でも、すべてがはるかに簡単になります。
Vegard Jervell

77

matplotlibバージョン2.1以降では、図の凡例を使用できます。ax.legend()Axesからのハンドルで凡例を生成する代わりに、axFigureの凡例を作成できます。

fig.legend(loc = "右上")

これは、図のすべてのサブプロットからすべてのハンドルを収集します。これは図の凡例であるため、図の隅に配置され、loc引数は図に関連しています。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0,10)
y = np.linspace(0,10)
z = np.sin(x/3)**2*98

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(x,y, '-', label = 'Quantity 1')

ax2 = ax.twinx()
ax2.plot(x,z, '-r', label = 'Quantity 2')
fig.legend(loc="upper right")

ax.set_xlabel("x [units]")
ax.set_ylabel(r"Quantity 1")
ax2.set_ylabel(r"Quantity 2")

plt.show()

ここに画像の説明を入力してください

凡例を軸に戻すには、a bbox_to_anchorとaを指定しbbox_transformます。後者は、凡例が存在する軸の座標軸変換です。前者は、loc与えられた軸座標によって定義されるエッジの座標である場合があります。

fig.legend(loc="upper right", bbox_to_anchor=(1,1), bbox_transform=ax.transAxes)

ここに画像の説明を入力してください


では、バージョン2.1はすでにリリースされていますか?しかし、Anaconda 3では、conda upgrade matplotlib新しいバージョンは見つかりませんでした。まだv.2.0.2を使用しています
StayFoolish

1
これは、最終結果を達成するためのよりクリーンな方法です。
ゴーサム

1
美しくてパイソン風
DanGoodrick

1
これは、多くのサブプロットがある場合は機能しないようです。すべてのサブプロットに単一の凡例を追加します。通常、サブプロットごとに1つの凡例が必要であり、各凡例の主軸と副軸の両方に系列が含まれています。
sancho.s ReinstateMonicaCellio

@sancho正解です。それが、この回答の3番目の文に書かれていることです。「図のすべてのサブプロットからすべてのハンドルが収集されます。」
ImportanceOfBeingErnest

38

あなたはaxに行を追加することで簡単にあなたが望むものを得ることができます:

ax.plot([], [], '-r', label = 'temp')

または

ax.plot(np.nan, '-r', label = 'temp')

これは、axの凡例にラベルを追加するだけです。

これははるかに簡単な方法だと思います。上記のように手作業で修正するのは非常に簡単なので、2番目の軸に数本の線しかない場合は、線を自動的に追跡する必要はありません。とにかく、それはあなたが必要とするものに依存します。

コード全体は次のとおりです。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('mathtext', default='regular')

time = np.arange(22.)
temp = 20*np.random.rand(22)
Swdown = 10*np.random.randn(22)+40
Rn = 40*np.random.rand(22)

fig = plt.figure()
ax = fig.add_subplot(111)
ax2 = ax.twinx()

#---------- look at below -----------

ax.plot(time, Swdown, '-', label = 'Swdown')
ax.plot(time, Rn, '-', label = 'Rn')

ax2.plot(time, temp, '-r')  # The true line in ax2
ax.plot(np.nan, '-r', label = 'temp')  # Make an agent in ax

ax.legend(loc=0)

#---------------done-----------------

ax.grid()
ax.set_xlabel("Time (h)")
ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")
ax2.set_ylabel(r"Temperature ($^\circ$C)")
ax2.set_ylim(0, 35)
ax.set_ylim(-20,100)
plt.show()

プロットは次のとおりです。

ここに画像の説明を入力してください


更新:より良いバージョンを追加:

ax.plot(np.nan, '-r', label = 'temp')

これはplot(0, 0)、軸の範囲を変更する可能性がある間は何もしません。


分散のもう1つの例

ax.scatter([], [], s=100, label = 'temp')  # Make an agent in ax
ax2.scatter(time, temp, s=10)  # The true scatter in ax2

ax.legend(loc=1, framealpha=1)

3
私はこれが好き。システムを「だます」方法で見た目は醜いですが、実装は簡単です。
Daniel Power

これは実装が本当に簡単です。しかし、これをスキャッターと一緒に使用すると、凡例の結果のスキャッターサイズはほんの小さなポイントになります。
greeeeeeen

@greeeeeeen次に、散布図を作成するときにマーカーのサイズを指定するだけです:-)
Syrtis Major

@SyrtisMajorもちろん私も試してみました。しかし、それでも凡例のマーカーサイズは変わりませんでした。
greeeeeeen

@greeeeeeenエージェント散布のマーカーサイズを変更しましたか?私の投稿を見て、サンプルコードのスニペットを追加しました。
シルティスメジャー

7

あなたのニーズに合うかもしれないクイックハック。

ボックスのフレームを外して、2つの凡例を隣り合わせに手動で配置します。このようなもの..

ax1.legend(loc = (.75,.1), frameon = False)
ax2.legend( loc = (.75, .05), frameon = False)

locタプルは、グラフの場所を表す左から右および下から上へのパーセンテージです。


5

host_subplotを使用して複数のy軸とすべての異なるラベルを1つの凡例に表示する以下の公式のmatplotlibの例を見つけました。回避策は必要ありません。これまでに見つけた最良の解決策。 http://matplotlib.org/examples/axes_grid/demo_parasite_axes2.html

from mpl_toolkits.axes_grid1 import host_subplot
import mpl_toolkits.axisartist as AA
import matplotlib.pyplot as plt

host = host_subplot(111, axes_class=AA.Axes)
plt.subplots_adjust(right=0.75)

par1 = host.twinx()
par2 = host.twinx()

offset = 60
new_fixed_axis = par2.get_grid_helper().new_fixed_axis
par2.axis["right"] = new_fixed_axis(loc="right",
                                    axes=par2,
                                    offset=(offset, 0))

par2.axis["right"].toggle(all=True)

host.set_xlim(0, 2)
host.set_ylim(0, 2)

host.set_xlabel("Distance")
host.set_ylabel("Density")
par1.set_ylabel("Temperature")
par2.set_ylabel("Velocity")

p1, = host.plot([0, 1, 2], [0, 1, 2], label="Density")
p2, = par1.plot([0, 1, 2], [0, 3, 2], label="Temperature")
p3, = par2.plot([0, 1, 2], [50, 30, 15], label="Velocity")

par1.set_ylim(0, 4)
par2.set_ylim(1, 65)

host.legend()

plt.draw()
plt.show()

Stack Overflowへようこそ!ターゲットサイトにアクセスできない、または完全にオフラインになる場合に備えて、リンクの最も関連性の高い部分を引用してください。参照してください。私は良い答えを書くにはどうすればよいです。今後のより多くの現在の質問に焦点を当て、これはほぼ4年前のものです。
ByteHamster

確かに良い発見ですが、例から学んだことを取り入れて、それをOPのMWEに適用し、画像を含めていただければ幸いです。
aeroNotAuto 2017年
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.