web-dev-qa-db-fra.com

Tracer plusieurs lignes avec pandas dataframe

J'ai un cadre de données qui ressemble à ce qui suit

   color  x   y
0    red  0   0
1    red  1   1
2    red  2   2
3    red  3   3
4    red  4   4
5    red  5   5
6    red  6   6
7    red  7   7
8    red  8   8
9    red  9   9
10  blue  0   0
11  blue  1   1
12  blue  2   4
13  blue  3   9
14  blue  4  16
15  blue  5  25
16  blue  6  36
17  blue  7  49
18  blue  8  64
19  blue  9  81

Je veux finalement deux lignes, une bleue et une rouge. La ligne rouge devrait être essentiellement y = x et la ligne bleue devrait être y = x ^ 2

Quand je fais ce qui suit:

df.plot(x='x', y='y')

La sortie est la suivante:

Existe-t-il un moyen de faire savoir à pandas qu'il y a deux ensembles? Et les regrouper en conséquence. J'aimerais pouvoir spécifier la colonne 'couleur' ​​en tant qu'ensemble différenciateur

38
sedavidw

Vous pouvez utiliser groupby pour scinder le DataFrame en sous-groupes en fonction de la couleur:

for key, grp in df.groupby(['color']):

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

df = pd.read_table('data', sep='\s+')
fig, ax = plt.subplots()

for key, grp in df.groupby(['color']):
    ax = grp.plot(ax=ax, kind='line', x='x', y='y', c=key, label=key)

plt.legend(loc='best')
plt.show()

les rendements enter image description here

57
unutbu

Un autre moyen simple consiste à utiliser la fonction pivot pour formater les données selon vos besoins.

df.plot() fait le reste

df = pd.DataFrame([
    ['red', 0, 0],
    ['red', 1, 1],
    ['red', 2, 2],
    ['red', 3, 3],
    ['red', 4, 4],
    ['red', 5, 5],
    ['red', 6, 6],
    ['red', 7, 7],
    ['red', 8, 8],
    ['red', 9, 9],
    ['blue', 0, 0],
    ['blue', 1, 1],
    ['blue', 2, 4],
    ['blue', 3, 9],
    ['blue', 4, 16],
    ['blue', 5, 25],
    ['blue', 6, 36],
    ['blue', 7, 49],
    ['blue', 8, 64],
    ['blue', 9, 81],
], columns=['color', 'x', 'y'])

df = df.pivot(index='x', columns='color', values='y')

df.plot()

result

pivot transforme efficacement les données en:

enter image description here

28
MrE

Si vous avez seaborn installé, une méthode plus simple qui ne vous oblige pas à exécuter pivot:

import seaborn as sns

sns.lineplot(data=df, x='x', y='y', hue='color')
7
Cheng