Mit mehreren Farben in matplotlib Grundstück
Habe ich ein numpy-array von 2D-Daten Punkte (x,y), die in drei Kategorien eingeteilt (0,1,2).
a = array([[ 1, 2, 3, 4, 5, 6, 7, 8 ],
[ 9, 8, 7, 6, 5, 4, 3, 2 ]])
class = array([0, 2, 1, 1, 1, 2, 0, 0])
Meine Frage ist, ob ich zeichnen können diese Punkte mit verschiedenen Farben. Ich würde gerne etwas wie das hier tun:
colors = list()
for i in class:
if i == 0:
colors.append('r')
elif i == 1:
colors.append('g')
else:
colors.append('b')
print colors
['r', 'b', 'g', 'g', 'g', 'b', 'r', 'r']
pp.plot(a[0], a[1], color = colors)
Du musst angemeldet sein, um einen Kommentar abzugeben.
Ich nehme an, Sie zeichnen möchten unterschiedliche Punkte. In diesem Fall
wenn Sie definieren ein numpy-array:
dann können Sie erzeugen das array von Farben mit
colormap[categories]
:Erträge