Python nuage de points de couleurs différentes en fonction de la valeur

J'ai un dataframe qui je veux faire un diagramme de dispersion.

le dataframe ressemble:

       year  length  Animation
0      1971     121       1
1      1939      71       1
2      1941       7       0
3      1996      70       1
4      1975      71       0

Je veux les points dans mon nuage de points pour être d'une couleur différente en fonction de la valeur dans l'Animation ligne.
Animation = 1 = jaune
animation = 0 = noir
ou quelque chose de similaire

J'ai essayé de faire ce qui suit:

dfScat = df[['year','length', 'Animation']]
dfScat = dfScat.loc[dfScat.length < 200]    
axScat = dfScat.plot(kind='scatter', x=0, y=1, alpha=1/15, c=2)

Il en résulte un curseur qui rend difficile de faire la différence. Python nuage de points de couleurs différentes en fonction de la valeur

OriginalL'auteur Rainoa | 2017-04-18

  1. 5

    Utiliser le c paramètre dans scatter

    df.plot.scatter('year', 'length', c='Animation', colormap='jet')

    Python nuage de points de couleurs différentes en fonction de la valeur

    C'est exactement ce que je fais. Je suis juste en utilisant leur index au lieu du nom. Bon heads up tho. Mais comment pourrais-je aller autour de la modification de la couleur complètement en fonction de leur valeur. Je n'ai rien entre 0 et 1. Par exemple que 1 = jaune et 0 = noir sur le curseur.
    Essayez d'utiliser n'nom de la colonne. Utiliser une autre palette de couleurs.
    Cheers, Ne savent pas à propos de carte des couleurs. 🙂

    OriginalL'auteur piRSquared

  2. 4

    Vous pouvez également affecter des couleurs discrètes pour les points de passage d'un tableau à c=
    Comme ceci:

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

d = {"year"      : (1971, 1939, 1941, 1996, 1975),
     "length"    : ( 121,   71,    7,   70,   71),
     "Animation" : (   1,    1,    0,    1,    0)}

df = pd.DataFrame(d)
print(df)

colors = np.where(df["Animation"]==1,'y','k')
df.plot.scatter(x="year",y="length",c=colors)
plt.show()

    Cela donne:

       Animation  length  year
0          1     121  1971
1          1      71  1939
2          0       7  1941
3          1      70  1996
4          0      71  1975

    Python nuage de points de couleurs différentes en fonction de la valeur

    OriginalL'auteur Arjaan Buijk