Was ist der Unterschied zwischen der Verwendung von `[data2]` und `[[data2]]` mit `groupby`

Question

Ich arbeite durch eine Python für Datenanalyse-Tutorial und möchte einige Erläuterungen zu der Ausgabe von [data2] und [[data2]] bei Verwendung von groupby.

Answer 1

Wenn Gebrauch:

[data2] 

Sie Series mit Multiindex bekommen.

Wenn Verwendung Teilmenge

[[data2]] 

Sie DataFrame mit Multiindex bekommen.

Und wenn Gebrauch:

df.groupby(['key1','key2'], as_index=False)['data2'].mean() 

Sie ohne MultiindexDataFrame mit 3 Spalten erhalten.

Vielleicht ist es klar, ob eine andere Form verwenden:

import pandas as pd 
df = pd.DataFrame({'key1':[1,2,2,1,2,2], 
        'key2':[4,4,4,4,5,5], 
        'data2':[7,8,9,1,3,5], 
        'D':[1,3,5,7,9,5]}) 

print (df) 
    D data2 key1 key2 
0 1  7  1  4 
1 3  8  2  4 
2 5  9  2  4 
3 7  1  1  4 
4 9  3  2  5 
5 5  5  2  5 


print (df['data2'].groupby([df.key1,df.key2]).mean()) 
key1 key2 
1  4  4.0 
2  4  8.5 
     5  4.0 
Name: data2, dtype: float64 

print (df[['data2']].groupby([df.key1,df.key2]).mean()) 
      data2 
key1 key2  
1 4  4.0 
2 4  8.5 
    5  4.0