Färben von Streudiagrammen nach Spaltenwerten in Python

Die Vielseitigkeit von ggplot2 in R ermöglicht die nahtlose Zuweisung von Farben zu Datenpunkten basierend auf Spalten Werte. Diese Funktion kann auch in Python mithilfe von Pandas-Datenrahmen und Matplotlib repliziert werden.

Verwenden von Pandas und Matplotlib

Berücksichtigen Sie die folgenden Schritte, um Farben Werten in Matplotlib zuzuordnen:

1. Erstellen Sie ein Farbwörterbuch: Definieren Sie ein Wörterbuch, das eindeutige Werte zuordnet in der kategorialen Spalte auf eine entsprechende Farbe. Dies stellt eine konsistente Farbzuweisung über alle Datenpunkte hinweg sicher.
2. Eine Farbspalte hinzufügen: Erstellen Sie eine neue Spalte im Datenrahmen, die jedem Wert in der kategorialen Spalte die entsprechende Farbe zuweist.
3. Zeichnen Sie das Streudiagramm: Verwenden Sie den c-Parameter in matplotlib.pyplot.scatter, um die Farbspalte als Farbe anzugeben Argument.

Hier ist eine Beispielimplementierung:

def dfScatter(df, xcol='Height', ycol='Weight', catcol='Gender'):
    fig, ax = plt.subplots()
    categories = np.unique(df[catcol])
    colors = np.linspace(0, 1, len(categories))
    colordict = dict(zip(categories, colors))  

    df["Color"] = df[catcol].apply(lambda x: colordict[x])
    ax.scatter(df[xcol], df[ycol], c=df.Color)
    return fig

Beispielverwendung

Betrachten Sie einen Datenrahmen mit den Spalten „Größe“, „Gewicht“ und „Geschlecht“. So erstellen Sie ein Streudiagramm, in dem Farben basierend auf der Spalte „Geschlecht“ zugewiesen werden:

df = pd.DataFrame({'Height':np.random.normal(size=10),
                       'Weight':np.random.normal(size=10),
                       'Gender': ["Male","Male","Unknown","Male","Male",
                                  "Female","Did not respond","Unknown","Female","Female"]})    
fig = dfScatter(df)

Dadurch wird ein Streudiagramm erstellt, in dem die Spalte „Geschlecht“ die Farbe jedes Datenpunkts bestimmt.