web-dev-qa-db-fra.com

Pandas groupe de données par trame)

J'ai un cadre de données structuré comme suit:

          Date   ticker  adj_close 
0   2016-11-21     AAPL    111.730     
1   2016-11-22     AAPL    111.800    
2   2016-11-23     AAPL    111.230    
3   2016-11-25     AAPL    111.790     
4   2016-11-28     AAPL    111.570    
...          
8   2016-11-21      ACN    119.680            
9   2016-11-22      ACN    119.480              
10  2016-11-23      ACN    119.820              
11  2016-11-25      ACN    120.740 
...

Comment puis-je tracer basé sur le ticker le adj_close versus Date?

pythonpandasmatplotlibdataframe
33
MCM

Parcelle simple,

vous pouvez utiliser:

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

Vous pouvez également définir l’index sur Date à l’avance. Il est alors facile de tracer la colonne de votre choix:

df.set_index('Date', inplace=True)
df['adj_close'].plot()

Si vous voulez un graphique avec une série de ticker dessus

Vous devez groupby avant:

df.set_index('Date', inplace=True)
df.groupby('ticker')['adj_close'].plot(legend=True)

enter image description here

Parcelle simple,

vous pouvez utiliser:

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

Vous pouvez également définir l’index sur Date à l’avance. Il est alors facile de tracer la colonne de votre choix:

df.set_index('Date', inplace=True)
df['adj_close'].plot()

Si vous voulez un graphique avec sous-parcelles individuelles:

grouped = df.groupby('ticker')

ncols=2
nrows = int(np.ceil(grouped.ngroups/ncols))

fig, axes = plt.subplots(nrows=nrows, ncols=ncols, figsize=(12,4), sharey=True)

for (key, ax) in Zip(grouped.groups.keys(), axes.flatten()):
    grouped.get_group(key).plot(ax=ax)

ax.legend()
plt.show()

enter image description here

55
Julien Marrec

Semblable à la réponse de Julien ci-dessus, j'ai eu du succès avec ce qui suit:

fig, ax = plt.subplots(figsize=(10,4))
for key, grp in df.groupby(['ticker']):
    ax.plot(grp['Date'], grp['adj_close'], label=key)

ax.legend()
plt.show()

Cette solution pourrait être plus pertinente si vous voulez plus de contrôle dans matlab.

Solution inspirée par: https://stackoverflow.com/a/52526454/10521959

2
Yaakov Bressler