2013-09-04 22 views
5

Usando i panda, è possibile calcolare una singola tabella a tabelle incrociate (o tabella pivot) contenente valori calcolati da due diverse funzioni?come utilizzare due diverse funzioni all'interno di crosstab/pivot_table nei panda?

import pandas as pd 
import numpy as np 

c1 = np.repeat(['a','b'], [50, 50], axis=0) 
c2 = list('xy'*50) 
c3 = np.repeat(['G1','G2'], [50, 50], axis=0) 
np.random.shuffle(c3) 
c4=np.repeat([1,2], [50,50],axis=0) 
np.random.shuffle(c4) 
val = np.random.rand(100) 

df = pd.DataFrame({'c1':c1, 'c2':c2, 'c3':c3, 'c4':c4, 'val':val}) 

frequencyTable = pd.crosstab([df.c1,df.c2],[df.c3,df.c4]) 
meanVal = pd.crosstab([df.c1,df.c2],[df.c3,df.c4],values=df.val,aggfunc=np.mean) 

Quindi, sia le righe e le colonne sono gli stessi in entrambe le tabelle, ma quello che mi piacerebbe davvero è un tavolo con due frequenze e valori medi:

c3   G1      G2   
c4  1    2    1    2 
c1 c2 freq val  freq val  freq val  freq val   
a x 6 0.624931 5 0.582268 8 0.528231 6 0.362804 
    y 7 0.493890 8 0.465741 3 0.613126 7 0.312894 
b x 9 0.488255 5 0.804015 6 0.722640 5 0.369480 
    y 6 0.462653 4 0.506791 5 0.583695 10 0.517954 
+0

Hai provato a utilizzare "aggfunc = [func1, func2]'? Non ho idea se questo funziona, ma penso di averlo visto prima da qualche parte. – Brian

+0

@Brian: Sì, ho appena provato la risposta di joris e funziona benissimo! Grazie molto! – HappyPy

+1

Dovrei solo iniziare a postare i miei commenti come risposte :-p – Brian

risposta

10

si può dare una elenco delle funzioni:

pd.crosstab([df.c1,df.c2], [df.c3,df.c4], values=df.val, aggfunc=[len, np.mean]) 

Se si desidera che la tabella come mostrato nella sua interrogazione, si dovrà riorganizzare i livelli un po ':

In [42]: table = pd.crosstab([df.c1,df.c2], [df.c3,df.c4], values=df.val, aggfunc=[len, np.mean]) 

In [43]: table 
Out[43]: 
     len    mean        
c3  G1  G2   G1     G2   
c4  1 2 1 2   1   2   1   2 
c1 c2              
a x  4 6 8 7 0.303036 0.414474 0.624900 0.425234 
    y  5 5 8 7 0.543363 0.480419 0.583499 0.637657 
b x 10 6 4 5 0.400279 0.436929 0.442924 0.287572 
    y  6 8 5 6 0.400427 0.623319 0.764506 0.408708 

In [44]: table.reorder_levels([1, 2, 0], axis=1).sort_index(axis=1) 
Out[44]: 
c3  G1       G2       
c4  1    2    1    2   
     len  mean len  mean len  mean len  mean 
c1 c2                
a x  4 0.303036 6 0.414474 8 0.624900 7 0.425234 
    y  5 0.543363 5 0.480419 8 0.583499 7 0.637657 
b x 10 0.400279 6 0.436929 4 0.442924 5 0.287572 
    y  6 0.400427 8 0.623319 5 0.764506 6 0.408708 
+0

Wow, è così semplice :) Molte grazie! – HappyPy

+0

BTW, è possibile rendere le colonne "len" e "media" quando si rimodellano con la fusione? Ho provato: pd.melt (table.reset_index(), id_vars = ['c1', 'c2']), ma ottengo una colonna denominata "NaN" con "len" e "mean" come elementi. Grazie! – HappyPy

+1

Cosa intendi esattamente con "colonne proprie"? Vuoi una colonna con "mean" e una colonna con "len" affissa le colonne "c1", "c2", "c3", "c4", come si otterrebbe con 'df.groupby ([" c1 ", "c2", "c3", "c4"]). aggregato ([len, np.mean]). reset_index() '? In tal caso puoi farlo direttamente invece che tramite crosstab e fusione. – joris