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
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
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()
pivot transforme efficacement les données en:
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')