387

TL;DR ->追加の変数を作成せずに、Matplotlibの折れ線グラフの凡例を作成するにはどうすればよいですか?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()

ご覧のとおり、これは の非常に基本的な使い方matplotlibですPyPlot。これにより、理想的には次のようなグラフが生成されます。

グラフ

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

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)

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

4

7 に答える 7

651

label=各呼び出しに a を追加してからplot()、 を呼び出します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

于 2013-10-01T21:00:59.147 に答える
9

カスタム凡例ドキュメントを追加できます

first = [1, 2, 4, 5, 4]
second = [3, 4, 2, 2, 3]
plt.plot(first, 'g--', second, 'r--')
plt.legend(['First List', 'Second List'], loc='upper left')
plt.show()

ここに画像の説明を入力

于 2019-07-12T17:00:16.307 に答える