質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

88.32%

すでに描画されているグラフに凡例(ラベル)をつける方法について

解決済

回答 3

投稿 編集

  • 評価
  • クリップ 0
  • VIEW 786

harunouta

score 144

前提・実現したいこと

すでに描画されているグラフに凡例(ラベル)をつけようとしています。
具体的には該当のソースコードのオレンジ色の線に「No.1」、緑色の線に「No.2」という具合です。

発生している問題・エラーメッセージ

以下のようなグラフをプロットしたのですが、データごとにラベルを付与してプロットしたわけではないので、グラフを描画した後で、凡例を付与する方法がわかりません。

該当のソースコード

import matplotlib.pyplot as plt
data = [[1,2,4,3,5,2],
       [1,1,4,2,6,3],
       [0, 2,5, 2,6,2]]
plt.plot(data)
plt.title("sample")
plt.show()


出力グラフ

試したこと

参考記事にあるような、データをプロットする前に凡例名をつける方法はわかりましたが、今回のプログラムではデータ別に指定していないので、この方法は適用できません。

plt.title("moon")
plt.scatter(x0, y0, label="label-A")
plt.scatter(x1, y1, label="label-B")

plt.xlabel("X-LABEL")
plt.xlabel("Y-LABEL")
plt.legend()

補足情報(FW/ツールのバージョンなど)

python3.6

ご回答を受けての追記

dataが以下の変形を受けて作成された場合、凡例を付与するためのコードを実行しても出力が異なります。
型はどちらもlistですが、なぜ出力がちがうのか、またこのような変形を経て生成する場合のプロットはどのように修正すべきでしょうか。

data = [1,2,4,3,5,2, 1,1,4,2,6,3, 0, 2,5, 2,6,2]
data = np.array(data)
data = data.reshape(-1, 6)

data
"""
array([[1, 2, 4, 3, 5, 2],
       [1, 1, 4, 2, 6, 3],
       [0, 2, 5, 2, 6, 2]])
"""
fig, ax = plt.subplots()
ax.plot(data)
ax.set_title("sample")

# 凡例を設定する。
labels = ["A", "B", "C", "D", "E"]
ax.legend(ax.lines, labels)

plt.plot()

#出力
[]
  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 3

checkベストアンサー

+1

matplotlib.axes.Axes.legend による凡例の作成方法には、以下の方式がありますので、これをお使いください。

legend(Artist のリスト, ラベルのリスト)

Artist は matplotlib で描画されるオブジェクトのことで、今回関心がある折れ線は Axes.lines で取得できます。

print(ax.lines)
# [<matplotlib.lines.Line2D at 0x7f5a3983f5c0>,
#  <matplotlib.lines.Line2D at 0x7f5a39781518>,
#  <matplotlib.lines.Line2D at 0x7f5a39781668>,
#  <matplotlib.lines.Line2D at 0x7f5a397817b8>,
#  <matplotlib.lines.Line2D at 0x7f5a39781908>,
#  <matplotlib.lines.Line2D at 0x7f5a39781a58>]

この各々が1つの折れ線を表す Artist オブジェクトなので、それに対応するラベル一覧を legend() の第2引数に渡してください。

コード全体

import matplotlib.pyplot as plt

data = np.array([1, 2, 4, 3, 5, 2, 1, 1, 4, 2, 6, 3, 0, 2, 5, 2, 6, 2])
data = data.reshape(-1, 6)
print(data)
# [[1 2 4 3 5 2]
#  [1 1 4 2 6 3]
#  [0 2 5 2 6 2]]

fig, ax = plt.subplots()
ax.plot(data)
ax.set_title("sample")

# 凡例を設定する。
labels = ["A", "B", "C", "D", "E", "F"]
ax.legend(ax.lines, labels)

plt.show()

イメージ説明

matplotlib - legend() を使った凡例のカスタマイズ方法 まとめ

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2019/06/09 16:09

    #出力
    []
    と追記内で明記いたしましたが、わかりにくくて申し訳ございません。

    キャンセル

  • 2019/06/09 16:10

    すいません。コードの最後が plt.plot() となっていたのですが、plt.show() でした。
    plt.show() でグラフが表示されると思います。
    回答のコードを修正しました。

    キャンセル

  • 2019/06/09 16:15

    解決いたしました。
    ありがとうございました。

    キャンセル

+1

ラベルの扱いとか面倒臭いので、個人的にはpandas使ってプロットすると楽だと思います。
このようなコードはいかがでしょうか。

import matplotlib.pyplot as plt
import pandas as pd


df = pd.DataFrame({ 'label-A' : [1,2,4,3,5,2],'label-B' : [1,1,4,2,6,3]})
df.plot()

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

+1

axis.linesで各線がとれるので、それらに対しset_labelすることができます。

import matplotlib.pyplot as plt
data = [[1,2,4,3,5,2],
       [1,1,4,2,6,3],
       [0, 2,5, 2,6,2]]
plt.plot(data)

axis = plt.gca()
for idx, line in enumerate(axis.lines):
    line.set_label('No.{}'.format(idx+1))
plt.legend()

plt.title("sample")
plt.show()


イメージ説明

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 88.32%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る