Zeichnen Sie Punkte mit Matplotlib.pyplot [[x1, y1], [x2, y2]]

Question

Ich möchte Graphen mit einer Liste von (x,y) Paare anstelle von zwei Listen, eines von X und eines von Y's zu zeichnen. Etwas wie folgt aus:

a = [[1,2],[3,3],[4,4],[5,2]] 
plt.plot(a, 'ro') 

Statt:

plt.plot([1,3,4,5], [2,3,4,2]) 

Verbesserungsvorschläge?

Answer 1

Schreiben Sie eine Hilfsfunktion.

Hier ist eine längliche Version, aber ich bin sicher, dass es einen Trick gibt, es zu komprimieren.

>>> def helper(lst): 
    lst1, lst2 = [], [] 
    for el in lst: 
     lst1.append(el[0]) 
     lst2.append(el[1]) 
    return lst1, lst2 

>>> 
>>> helper([[1,2],[3,4],[5,6]]) 
([1, 3, 5], [2, 4, 6]) 
>>> 

Auch diese Helfer hinzu:

def myplot(func, lst, flag): 
    return func(helper(lst), flag) 

Und es wie so nennen:

myplot(plt.plot, [[1,2],[3,4],[5,6]], 'ro') 

Alternativ Sie eine Funktion zu einem bereits instanzierte Objekt hinzufügen könnte.

Answer 2

Sie können etwas tun:

a=[[1,2],[3,3],[4,4],[5,2]] 
plt.plot(*zip(*a)) 

Leider kann man nicht mehr 'ro' übergeben. Sie müssen Marker und Linienstil Werte als Schlüsselwort-Parameter übergeben:

a=[[1,2],[3,3],[4,4],[5,2]] 
plt.plot(*zip(*a), marker='o', color='r', ls='') 

Der Trick, den ich verwenden unpacking argument lists ist.

Answer 3

Wenn Sie numpy Array verwenden Sie Achse extrahieren könnte:

a = array([[1,2],[3,3],[4,4],[5,2]]) 
plot(a[:,0], a[:,1], 'ro') 

Für Listen oder Listen Sie einige Helfer benötigen, wie:

a = [[1,2],[3,3],[4,4],[5,2]] 
plot(*sum(a, []), marker='o', color='r') 

Answer 4

Listenkomprehensionen

Ich schlage die liberale Anwendung von Listenergänzungen vor. Sie sind nicht nur kurz und bündig, sie machen auch Ihren Code sehr lesbar.

versuchen, etwas wie folgt aus:

list_of_lists = [[1,2],[3,3],[4,4],[5,2]]  
x_list = [x for [x, y] in list_of_lists] 
y_list = [y for [x, y] in list_of_lists] 

plt.plot(x_list, y_list) 

Argument auspacken sollte vermieden werden. Es ist hässlich.