[Debutant] Matplotlib - Colors/pie

Version imprimable

Salut,

J ai un petit probleme. Je souhaite tracer plusieurs pie charts avec pour tous la meme couleur pour le meme element. Je m explique:

Par exemple dans le code suivant, je trace 2 graphes. Mon probleme est que l element Frogs est bien bleu dans les deux graphes car c'est le premier element de mes series, mais Dogs qui est une foi 3ieme une foi second n a pas la meme couleur ...
Code:

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 from pylab import * labels = 'Frogs', 'Hogs', 'Dogs', 'Logs' fracs = [15,30,45, 10] #colors= ('b', 'g', 'r', 'c') explode=(0, 0.05, 0, 0) subplot(211) pie(fracs, explode=None, labels=labels, colors=None, autopct='%1.1f%%', shadow=True) title('Raining Hogs and Dogs', bbox={'facecolor':'0.8', 'pad':5}) labels = 'Frogs', 'Dogs' fracs = [60,40] subplot(212) pie(fracs, explode=None, labels=labels, colors=None, autopct='%1.1f%%', shadow=True) title('Raining Hogs and Dogs', bbox={'facecolor':'0.8', 'pad':5}) show()
resultat :
http://img15.imageshack.us/img15/2369/pielg.th.jpg

Je ne voudrais eviter d avoir a definir la sequence des couleurs manuellement pour chacun de mes graphes, mais definir en amont une sorte de legende du style :
Quand dans les grpahes c dogs utiliser bleu, quand c Frogs, utiliser vert ...

Est ce possible ?

Merci

Slumpy

11/02/2011, 21h47
Matthieu Brucher

Non, ce n'est pas possible.
12/02/2011, 19h38
rambc

C'est un peu dommage ce genre de limitations...
12/02/2011, 21h44
Matthieu Brucher

Les labels sont indépendants des valeurs stockées, donc ce n'est pas vraiment une limitation. Faire en sorte qu'un système marche, c'est pas immédiat du tout, et l'utilité est somme toute réduite à ceux qui ne veulent pas écrire une ligne supplémentaire (my_colors = [colors[label] for label in labels]).

Salut,

Matthieu, je suis pas sur de comprendre ca :

Citation:

(my_colors = [colors[label] for label in labels]).

Tu pourrais expliquer stp ?

Voila ce que j ai fait ... Ce n est vraiment pas terrible, mais au moins ca me permet d avoir la meme couleur pour un meme label sur chacun de mes graphes.

En gros, je repertorie des le depart tous les labels dont j aurai besoin et je les met dans un dictionnaire en leur attribuant une couleur.

Puis au moment d appeler mon graphe, je reconstruie la sequence de mes couleurs utilisant ce dictionnaire.

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
class aWRZ(object):
    def __init__(self):
        self.dico_LexD={}
 
    def addLexD(self, NewLexD,NewPerc):
        self.dico_LexD[NewLexD]=NewPerc
 
 
 
 
def PlotPie(aData,aLabel,acolor, aTitle):
    # make a square figure and axes
    fig=figure(1, figsize=(10,10))
    axes([0.2, 0.2, 0.5, 0.5])
 
    labels = aLabel
    fracs = aData
    colors= acolor
 
    #explode=(0, 0.0005, 0, 0)
    matplotlib.rcParams['font.size'] = 7
 
    pie(fracs, explode=None,labels=labels,
        colors=colors,
        autopct='%1.1f%%', shadow=True)
 
    title(aTitle, bbox={'facecolor':'0.8', 'pad':5})
    fig.savefig(aTitle + 'aquifer.png') #,dpi=100)
    close()
 
 
 
import csv
from pylab import *
import numpy
import random as  rn
 
dico_WRZ={}
dico_Legend={}
cr = csv.reader(open("C:/Book1.csv","rb"))
for a_CSV_row in cr:
    WC=a_CSV_row[0]
    WRZ=a_CSV_row[1]
    LexD=a_CSV_row[2]
    Perc=a_CSV_row[3]
    if dico_WRZ.has_key(WRZ)==False:
        dico_WRZ[WRZ]=aWRZ()
    dico_WRZ[WRZ].addLexD(LexD,Perc)
    #Colors
    if dico_Legend.has_key(LexD)==False:
        # create hex random color string 
        rcolor = '#' + "".join(["%02x"%rn.randrange(256) for x in range(3)])
        print rcolor
        dico_Legend[LexD]=rcolor 
 
for WRZkey in dico_WRZ.keys():
    print
    print WRZkey
    Labels=()
    Data=()
    Colors=()
    Title=WRZkey
    for Lexkey in dico_WRZ[WRZkey].dico_LexD.keys():
        Labels += (Lexkey,)
        Data += (dico_WRZ[WRZkey].dico_LexD[Lexkey],)
        Colors += (dico_Legend[Lexkey],)
        print Lexkey, dico_WRZ[WRZkey].dico_LexD[Lexkey]
    PlotPie(Data,Labels,Colors,Title)

Petite question:
Est ce que quelqu un saurait comment utilisant cette facon de faire, je pourrais utiliser une plage de couleur plutot que tirer des couleurs au hasard ? (en clair, pouvoir utiliser la rampe 'hot' ou autre). Donc modifier :

Code:

1
2
3
4
5
if dico_Legend.has_key(LexD)==False:
        # create hex random color string 
        rcolor = '#' + "".join(["%02x"%rn.randrange(256) for x in range(3)])
        print rcolor
        dico_Legend[LexD]=rcolor

Merci

Slumpy

14/02/2011, 14h18
rambc

Bonjour.

Citation:

Envoyé par Slumpy

Est ce que quelqu un saurait comment utilisant cette facon de faire, je pourrais utiliser une plage de couleur plutot que tirer des couleurs au hasard ? (en clair, pouvoir utiliser la rampe 'hot' ou autre).

Ce genre de choses se fait via une simple approximation affine. Par exemple, on part d'une couleur RGB (a,b,c) vers une autre (x,y,z). On définit alors la couleur intermédiaire de niveau t% par [f(a,x,t) , f(b,y,t) , f(c,z,t)] où f est la fonction définie par f(a,x,t) = t/100*(x-a) + a.

Citation:

Envoyé par Slumpy

Matthieu, je suis pas sur de comprendre ca : Tu pourrais expliquer stp ?

Grosso modo, c'est comme pour ta classe, sauf que c'est plus réduit comme écriture.

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
dico_WRZ={}
dico_Legend={}
cr = csv.reader(open("C:/Book1.csv","rb"))
for a_CSV_row in cr:
    WC=a_CSV_row[0]
    WRZ=a_CSV_row[1]
    LexD=a_CSV_row[2]
    Perc=a_CSV_row[3]
    if dico_WRZ.has_key(WRZ)==False:
        dico_WRZ[WRZ]={}
    dico_WRZ[WRZ][LexD] = Perc
    #Colors
    if dico_Legend.has_key(LexD)==False:
        # create hex random color string 
        rcolor = '#' + "".join(["%02x"%rn.randrange(256) for x in range(3)])
        print rcolor
        dico_Legend[LexD]=rcolor

Donc j'ai juste changé dico_WRZ[WRZ].

Maintenant, ta grosse fonction for devient :

Code:

1
2
3
4
5
6
7
8
9
10
11
for WRZkey in dico_WRZ.keys():
    print
    print WRZkey
    Labels=()
    Data=()
    Colors=()
    Title=WRZkey
    Labels = dico_WRZ[WRZkey].keys()
    Data = dico_WRZ[WRZkey].values()
    Colors = [dico_Legend[Lexkey] for Lexkey in Labels ]
    PlotPie(Data,Labels,Colors,Title)