Wie weise ich mehrere Labels gleichzeitig in Matplotlib zu?

Ich habe folgenden Datensatz:

x = [0, 1, 2, 3, 4]
y = [ [0, 1, 2, 3, 4],
      [5, 6, 7, 8, 9],
      [9, 8, 7, 6, 5] ]

Ich jetzt zeichnen Sie es mit:

import matplotlib.pyplot as plt
plt.plot(x, y)

Allerdings möchte ich zur Bezeichnung der 3 y-Datensätze mit diesem Befehl, der einen Fehler auslöst, wenn .legend() heißt:

lineObjects = plt.plot(x, y, label=['foo', 'bar', 'baz'])
plt.legend()

File "./plot_nmos.py", line 33, in <module>
  plt.legend()
...
AttributeError: 'list' object has no attribute 'startswith'

Wenn ich überprüfen das lineObjects:

>>> lineObjects[0].get_label()
['foo', 'bar', 'baz']
>>> lineObjects[1].get_label()
['foo', 'bar', 'baz']
>>> lineObjects[2].get_label()
['foo', 'bar', 'baz']

Frage

Gibt es einen eleganten Weg zu weisen mehrere labels, die nur mit der .plot() Methode?

InformationsquelleAutor der Frage Kit | 2012-07-14

33

Ist es nicht möglich, in den plot dieser beiden arrays gegen einander unmittelbar (mit mindestens version 1.1.1), daher müssen Sie sich in einer Schleife über deine y-arrays. Mein Rat wäre, um eine Schleife über die Beschriftungen in der gleichen Zeit:
```
import matplotlib.pyplot as plt

x = [0, 1, 2, 3, 4]
y = [ [0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [9, 8, 7, 6, 5] ]
labels = ['foo', 'bar', 'baz']

for y_arr, label in zip(y, labels):
    plt.plot(x, y_arr, label=label)

plt.legend()
plt.show()
```
Edit: @gcalmettes wies darauf hin, dass als numpy-arrays ist es möglich, in den plot alle Zeilen gleichzeitig (durch Umsetzung). Siehe @gcalmettes Antwort & Kommentare für details.

InformationsquelleAutor der Antwort pelson
35

Können Sie Durchlaufen Ihre Linie Objekte-Liste, so dass Etiketten sind einzeln zugewiesen. Ein Beispiel mit der eingebauten python- iter Funktion:
```
lineObjects = plt.plot(x, y)
plt.legend(iter(lineObjects), ('foo', 'bar', 'baz'))`
```
Edit: nach dem Update auf die matplotlib 1.1.1, es sieht aus wie die plt.plot(x, y) mit y-als Liste von Listen (soweit vom Autor nicht in Frage), funktioniert nicht mehr. Der Schritt, Plotten ohne iteration über die y-arrays ist immer noch möglich, dachte nach bestandener y als numpy.array (unter der Annahme, (numpy)[http://numpy.scipy.org/] wie zuvor eingeführt).

In diesem Fall verwenden Sie plt.plot(x, y) (wenn die Daten in der 2D-y-array angeordnet sind, wie Spalten, [1. Achse]) oder plt.plot(x, y.transpose()) (wenn die Daten in der 2D-y-array angeordnet sind, als Zeilen [Achse 0])

Edit 2: wie schon von @pelson (siehe Kommentar unten), die iter Funktion ist unnötig und einfach plt.legend(lineObjects, ('foo', 'bar', 'baz')) funktioniert perfekt

InformationsquelleAutor der Antwort gcalmettes

Kam ich über das gleiche problem und jetzt habe ich eine Lösung gefunden, dass ist den meisten einfach! Hoffentlich ist das nicht zu spät für Sie. Kein iterator, nur weisen Sie Ihr Ergebnis auf eine Struktur...

from numpy import *
from matplotlib.pyplot import *
from numpy.random import *

a = rand(4,4)
a
>>> array([[ 0.33562406,  0.96967617,  0.69730654,  0.46542408],
   [ 0.85707323,  0.37398595,  0.82455736,  0.72127002],
   [ 0.19530943,  0.4376796 ,  0.62653007,  0.77490795],
   [ 0.97362944,  0.42720348,  0.45379479,  0.75714877]])

[b,c,d,e] = plot(a)
legend([b,c,d,e], ["b","c","d","e"], loc=1)
show()

Sieht wie folgt aus:
Wie weise ich mehrere Labels gleichzeitig in Matplotlib zu?