凡例を可能な限り簡単な方法でMatplotlibのPyPlotに追加する


263

TL; DR - >どのようにして中線グラフの凡例を作成することができますMatplotlibs「をPyPlot余分な変数を作成せずに?

以下のグラフ化スクリプトを検討してください:

if __name__ == '__main__':
    PyPlot.plot(total_lengths, sort_times_bubble, 'b-',
                total_lengths, sort_times_ins, 'r-',
                total_lengths, sort_times_merge_r, 'g+',
                total_lengths, sort_times_merge_i, 'p-', )
    PyPlot.title("Combined Statistics")
    PyPlot.xlabel("Length of list (number)")
    PyPlot.ylabel("Time taken (seconds)")
    PyPlot.show()

あなたが見ることができるように、これは非常に基本的な使用であるmatplotlibPyPlot。これにより、理想的には次のようなグラフが生成されます。

グラフ

特別なことは何もありません。しかし、どのデータがどこにプロットされているかは不明です(私はいくつかのソートアルゴリズムのデータ、所要時間に対する長さをプロットしようとしているので、どの線がどれであるかを人々に知ってもらいたいと思います)。したがって、伝説が必要ですが、以下の例(公式サイトから)を見てください。

ax = subplot(1,1,1)
p1, = ax.plot([1,2,3], label="line 1")
p2, = ax.plot([3,2,1], label="line 2")
p3, = ax.plot([2,3,1], label="line 3")

handles, labels = ax.get_legend_handles_labels()

# reverse the order
ax.legend(handles[::-1], labels[::-1])

# or sort them by labels
import operator
hl = sorted(zip(handles, labels),
            key=operator.itemgetter(1))
handles2, labels2 = zip(*hl)

ax.legend(handles2, labels2)

追加の変数を作成する必要があることがわかりますax。この追加の変数を作成し、現在のスクリプトの単純さを維持せずに、グラフに凡例を追加するにはどうすればよいですか?


余分な変数を作成することについてのあなたの懸念に混乱しています。とにかく、これらのオブジェクトを舞台裏で作成する必要があります。
tacaswell 2013年

1
@tcaswellまあ、私はそれらを和らげるようにしてみましょう。スクリプト全体が複雑になるため、余分な変数を作成したくありません。私はこれをたくさんの学生に教えようとしています、そして彼らはmatplotlib以前に使ったことがないので、物事をできるだけ単純にしたかったです。また、Robの回答を見ると、Webサイトに表示されている例よりもはるかに単純です。お役に立てば幸いです。
Games Brainiac 2013年

1
ステートマシンインターフェイスを使用すると、その多くが「魔法」で行われているため、長期的には理解が難しくなると私は主張します。また、規則は次のimport matplotlib.pyplot as plt代わりに使用することですPyPlot
tacaswell

回答:


441

label=それぞれのplot()呼び出しにa を追加してから、を呼び出しますlegend(loc='upper left')

次のサンプルを検討してください(Python 3.8.0でテスト済み):

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 20, 1000)
y1 = np.sin(x)
y2 = np.cos(x)

plt.plot(x, y1, "-b", label="sine")
plt.plot(x, y2, "-r", label="cosine")
plt.legend(loc="upper left")
plt.ylim(-1.5, 2.0)
plt.show()

ここに画像の説明を入力してください このチュートリアルから少し変更:http : //jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html


2
シリーズがプロットされた時点でラベルがわからない場合にこれを行う方法はありますか?それはすでにプロットされた後にシリーズにラベルを追加する方法ですか?または、凡例を表示する前にプレースホルダーラベルを変更する方法はありますか?
davidA 2017

13
plt.legend(loc='upper left')も機能pltimport matplotlib.pyplot as pltます。
Matt Kleinsmith、

@eric、これに注意してくれてありがとう。コードを更新しました。
Robᵩ

スタックオーバーフローが素晴らしく、良い答えが変わる可能性があるため賛成されました
eric

1
@davidAはい、あなたは、単にに文字列のリストを渡すことができますplt.legendplt.legend(['First Label', 'Second Label'])
Apollysはモニカ・サポート

36

Axesインスタンス(ax)には、でアクセスできますplt.gca()。この場合、使用できます

plt.gca().legend()

これを行うには、次の実際の例のようにlabel=、各plt.plot()呼び出しでキーワードを使用するか、ラベルを内のタプルまたはリストとして割り当てますlegend

import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(-0.75,1,100)
y0 = np.exp(2 + 3*x - 7*x**3)
y1 = 7-4*np.sin(4*x)
plt.plot(x,y0,x,y1)
plt.gca().legend(('y0','y1'))
plt.show()

pltGcaLegend

しかし、あなたはより多くのAxesのインスタンスにアクセスする必要がある場合は一度、私はそれを変数に保存するお勧めしないということax

ax = plt.gca()

ax代わりに呼び出しますplt.gca()


読む必要がなく、写真付きのコピー&ペースト回答!この回答はより多くの信用に値します
Gulzar

14

ここにあなたを助けるための例があります...

fig = plt.figure(figsize=(10,5))
ax = fig.add_subplot(111)
ax.set_title('ADR vs Rating (CS:GO)')
ax.scatter(x=data[:,0],y=data[:,1],label='Data')
plt.plot(data[:,0], m*data[:,0] + b,color='red',label='Our Fitting 
Line')
ax.set_xlabel('ADR')
ax.set_ylabel('Rating')
ax.legend(loc='best')
plt.show()

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


2
気になるところですが、なぜフィッティングラインがデータから離れているのですか?
Apollysはモニカ

13

凡例付きの正弦曲線と余弦曲線の簡単なプロット。

中古 matplotlib.pyplot

import math
import matplotlib.pyplot as plt
x=[]
for i in range(-314,314):
    x.append(i/100)
ysin=[math.sin(i) for i in x]
ycos=[math.cos(i) for i in x]
plt.plot(x,ysin,label='sin(x)')  #specify label for the corresponding curve
plt.plot(x,ycos,label='cos(x)')
plt.xticks([-3.14,-1.57,0,1.57,3.14],['-$\pi$','-$\pi$/2',0,'$\pi$/2','$\pi$'])
plt.legend()
plt.show()

正弦および余弦プロット(クリックして画像を表示)


6

それがグラフ化している系列に対応するプロット呼び出しの各引数にラベルを追加します。 label = "series 1"

次にPyplot.legend()、スクリプトの下部に追加するだけで、凡例にこれらのラベルが表示されます。


これは正しいアイデアですが、ラベルを追加しないので、凡例は空になります
tacaswell


0
    # Dependencies
    import numpy as np
    import matplotlib.pyplot as plt

    #Set Axes
    # Set x axis to numerical value for month
    x_axis_data = np.arange(1,13,1)
    x_axis_data

    # Average weather temp
    points = [39, 42, 51, 62, 72, 82, 86, 84, 77, 65, 55, 44]

    # Plot the line
    plt.plot(x_axis_data, points)
    plt.show()

    # Convert to Celsius C = (F-32) * 0.56
    points_C = [round((x-32) * 0.56,2) for x in points]
    points_C

    # Plot using Celsius
    plt.plot(x_axis_data, points_C)
    plt.show()

    # Plot both on the same chart
    plt.plot(x_axis_data, points)
    plt.plot(x_axis_data, points_C)

    #Line colors
    plt.plot(x_axis_data, points, "-b", label="F")
    plt.plot(x_axis_data, points_C, "-r", label="C")

    #locate legend
    plt.legend(loc="upper left")
    plt.show()
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.