Attribuer des couleurs de ligne dans les pandas

Je suis en train de tracer certaines données dans les pandas et intégré de la fonction plot idéalement parcelles d'une ligne par colonne. Ce que je veux faire, c'est attribuer manuellement chaque ligne d'une couleur basée sur une classification que je fais.

Les ouvrages suivants:

df = pd.DataFrame({'1': [1, 2, 3, 4], '2': [1, 2, 1, 2]})
s = pd.Series(['c','y'], index=['1','2'])
df.plot(color = s)

Mais quand mes indices sont des entiers, il ne fonctionne plus et le jette comme KeyError:

df = pd.DataFrame({1: [1, 2, 3, 4], 2: [1, 2, 1, 2]})
s = pd.Series(['c','y'], index=[1,2])
df.plot(color = s)

La façon dont je le comprends, c'est que quand un index entier est utilisé de toute façon il doit commencer à partir de 0. C'est mon sentiment depuis les travaux suivants: la

df = pd.DataFrame({0: [1, 2, 3, 4], 1: [1, 2, 1, 2]})
s = pd.Series(['c','y'], index=[1,0])
df.plot(color = s)

Ma question est:

  • Ce qui se passe ici?
  • En supposant que j'ai un index entier qui ne démarre pas à partir de 0 ou n'est pas formé d'une succession de numéros, comment puis-je faire ce travail sans avoir à convertir l'indice de chaîne ou de revisiter à partir de 0?

EDIT:

J'ai réalisé que, même dans le premier cas, le code ne fait pas ce que j'ai prévu de faire.
Il semble que les pandas correspond à l'indice de DataFrame et seulement si les deux sont des entiers indices en partant de 0. Si ce n'est pas le cas, un KeyError est jeté ou si l'index est un str l'ordre des éléments est utilisé.

Est-ce correct? Et est-il un moyen pour correspondre à la Série et DataFrame indices? Ou dois-je m'assurer de passer d'une liste de couleurs dans le bon ordre?

Vous ne voulez pas les indices de match. Vous êtes à la correspondance de la Série' l'Index pour le DataFrame de colonnes.

OriginalL'auteur GebitsGerbils | 2015-09-11

  1. 7

    Ce qui se passe ici?

    L'argument mot-clé couleur est héritée de matplotlib.pyplot.plot(). Les détails dans la documentation de ne pas préciser que vous pouvez mettre dans une liste de couleurs lors de la représentation. Étant donné que la couleur est un argument mot-clé à partir de matplotlib, je vous recommande de ne pas utiliser une Pandas Série pour contenir les valeurs de couleur.

    Comment puis-je faire ce travail?

    Utiliser une liste au lieu d'une Série. Si vous étiez à l'aide d'une Série avec un indice censé correspondre les colonnes de votre DataFrame de couleurs spécifiques, vous aurez besoin de trier la première Série. Si les colonnes ne sont pas dans l'ordre, vous devrez trier les colonnes ainsi.

    # Option 1
s = s.sort_index()
df.plot(color = s.values) # as per Fiabetto's answer

# Option 2
df.plot(color = ['c', 'y']) # other method
    J'ai fini le tri de la Série et le DataFrame puis tourna la Série dans une liste, comme l'a suggéré. Cela fonctionne sans problème, mais j'espérais qu'il y a une façon plus élégante, mais je peux voir comment il ne peut pas être si la couleur de mot-clé correspond à matplotlibs (qui, évidemment, ne sait rien à propos de la Série).
    Ouais, malheureusement, les Pandas est juste un wrapper de matplotlib. matplotlib permet à l'utilisateur de presque rien, mais il peut être vraiment salissant.
    Ligne de code dans l'option 2, manque une fermeture ].

    OriginalL'auteur thecircus

  2. 3

    Essayer:

    df.plot(color = s.values)

    cela va affecter les couleurs indépendamment de l'ampleur de l'index.

    EDIT:

    J'ai essayé avec trois colonnes:

    df = pd.DataFrame({'1': [1, 2, 3, 4], '2': [1, 2, 1, 2], '3': [4, 3, 2, 1]})
s = pd.Series(['c','y','r'], index=[1,3,2])
df.plot(color = s.sort_index().values)

    et de tri de la Série, il fonctionne.

    Cela fonctionne en effet, mais j'ai réalisé qu'en fait le problème semble être l'ordre. Je vais modifier la question de l'inclure!
    Le tri de la Série n'est utile que si le DataFrame de colonnes sont dans l'ordre.

    OriginalL'auteur Fabio Lamanna