ich will mit dieser Funktion einen Plot erstellen wo Punkte wenn das Label 1 ist rot angezeigt werden und wenn das Label -1 ist blau angezeigt werden. Das Problem ist wenn der letzte Punkt als Label 1 hat werden alle Punkte Rot angezeigt und wenn das Label -1 ist werden alle Punkte blau angezeigt. Was muss ich hier anderst machen das jeder Punkt die Farbe bekommt die seinem Label zugeordnet ist?
Code: Alles auswählen
def show(x,y,label) :
fig, ax = plt.subplots(figsize=(5,5))
for l in label:
if l == 1:
color='red'
ax.scatter(x, y, c=color, label=color)
elif l == -1:
color='blue'
ax.scatter(x, y, c=color, label=color)
ax.legend()
plt.show()