2017-02-14 1 views
2

J'ai tracé un graphique (ci-joint) en utilisant le code python ci-dessous. Le problème est que la couleur attribuée est très difficile pour les gens de faire la différence entre chaque point de données. Je voudrais personnaliser l'échelle de couleur. Quelqu'un peut-il aider avec ceci? Merci beaucoup!Comment personnaliser python DataFrame scatter plot couleur?

import pandas as pd 
import matplotlib.pyplot as plt 
%matplotlib inline 
x=[1,2,3,4,5,1,2,3,4,5] 
y=[5,4,3,2,1,1,2,3,4,5] 
z=[0.1,0.2,-0.1,0.3,0.05,0.1,-0.1,-0.5,0.25,-0.05] 
df=pd.DataFrame({'x':x,'y':y,'z':z}) 
df.plot.scatter(x='x',y='y',c='z') 

enter image description here

Répondre

3

Pass un argument colormap-df.plot(), par exemple,

df.plot.scatter(x='x',y='y',c='z', colormap='plasma') 

enter image description here

De la documentation pour pandas.DataFrame.plot:

colormap: str ou matplotlib objet colormap, par défaut Aucun

Colormap pour sélectionner les couleurs à partir. Si chaîne, chargez la colormap avec ce nom de matplotlib.

Vous pouvez consulter les cartes de couleurs matplotlib (comme j'ai utilisé ci-dessus) dans le colormaps_reference.