Erstellung von scatter_matrix

Question

Ich möchte ein ähnliches Diagramm wie das veröffentlichte here erhalten. In meinem Datensatz habe ich 3 Funktionen und 1 binäres Ziel. Wenn das Ziel = 0 ist, sollten die Punkte z.B. in rot. Wenn das Ziel = 1 ist, sollten die Punkte z.B. in Blau.

Wenn ich diesen Code ausführen, sieht es aus wie alle Punkte die gleiche Farbe haben:

from pandas.tools.plotting import scatter_matrix 
scatter_matrix(df, alpha=0.2, figsize=(6, 6), diagonal='kde') 

Wie ich verschiedene Farben angeben, für Sollwerte 0 und 1, so dass ich mögliche Gruppierungen beobachten kann?

Answer 1

Um diese Daten mit den pandasplotting Tools zu plotten, wie Sie jetzt sind, übergeben Sie color=['blue' if target == 1 else 'red'] als Argument.

Allerdings empfehle ich stattdessen die seabornpairplot Einrichtung verwenden. Dies führt zu einem Diagramm, das dem Tutorial sehr viel näher kommt, und um basierend auf einem Datenpunkt zu kolorieren, müssen Sie nur hue='my_data_column' als Parameter angeben. Cf. here.