Pandas - Streudiagramm mit verschiedener Farb-Legende für jeden Punkt

Question

aus dem folgende Beispiel starten:

fig, ax = plt.subplots() 

df = pd.DataFrame({'n1':[1,2,1,3], 'n2':[1,3,2,1], 'l':['a','b','c','d']}) 

for label in df['l']: 

    df.plot('n1','n2', kind='scatter', ax=ax, s=50, linewidth=0.1, label=label) 

, was ich erhielt die folgenden Scatterplot:

ich jetzt versuchen, Legen Sie für jeden der vier Punkte eine andere Farbe fest. Ich weiß, dass ich kann Schleife über eine Reihe von zum Beispiel vier Farben in einer Liste wie:

colorlist = ['b','r','c','y'] 

aber da meine reale Datenmenge mindestens 20 verschiedene Punkte umfasst, wurde ich für eine Art „Farbgenerator suche "um darin zu schleifen.

Answer 1

Die folgende Methode eine Liste von Farben, solange Ihre Datenrahmen erstellen, und dann Zeichnen Sie einen Punkt mit einem Etikett mit jeder Farbe:

import matplotlib.pyplot as plt 
import matplotlib.cm as cm 
import matplotlib.colors as colors 
import numpy as np 
import pandas as pd 

fig, ax = plt.subplots() 

df = pd.DataFrame({'n1':[1,2,1,3], 'n2':[1,3,2,1], 'l':['a','b','c','d']}) 

colormap = cm.viridis 
colorlist = [colors.rgb2hex(colormap(i)) for i in np.linspace(0, 0.9, len(df['l']))] 

for i,c in enumerate(colorlist): 

    x = df['n1'][i] 
    y = df['n2'][i] 
    l = df['l'][i] 

    ax.scatter(x, y, label=l, s=50, linewidth=0.1, c=c) 

ax.legend() 

plt.show() 

Answer 2

Wie wäre es damit,

---

Hier ist der Quellcode,

import matplotlib.pyplot as plt 
import pandas as pd 
import numpy as np 
from matplotlib import cm 

fig, ax = plt.subplots() 

df = pd.DataFrame({'n1':[1,2,1,3], 'n2':[1,3,2,1], 'l':['a','b','c','d']}) 

#colors = ['b','r','c','y'] 
nrof_labels = len(df['l']) 
colors = cm.rainbow(np.linspace(0, 1, nrof_labels))  # create a bunch of colors 

for i, r in df.iterrows(): 
    ax.plot(r['n1'], r['n2'], 'o', markersize=10, color=colors[i], linewidth=0.1, label=r['l']) 

ax.set_xlim(0.5, 3.5) 
ax.set_ylim(0.5, 3.5) 
plt.legend(loc='best') 

plt.show() 

Answer 3

IIUC Sie es auf diese Weise tun können:

import matplotlib.pyplot as plt 
from matplotlib import colors 
import pandas as pd 

colorlist = list(colors.ColorConverter.colors.keys()) 
fig, ax = plt.subplots() 
[df.iloc[[i]].plot.scatter('n1', 'n2', ax=ax, s=50, label=l, 
         color=colorlist[i % len(colorlist)]) 
for i,l in enumerate(df.l)] 

color:

In [223]: colorlist 
Out[223]: ['m', 'b', 'g', 'r', 'k', 'y', 'c', 'w'] 

PS colorlist[i % len(colorlist)] - sollte in der Liste immer bleiben begrenzt