2015-09-29 63 views
5

我試圖讓.describe()函數以重新格式化的方式輸出。 這裏是CSV數據(testProp.csv熊貓python .describe()格式化/輸出

'name','prop' 
A,1 
A,2 
B, 4 
A, 3 
B, 5 
B, 2 

當我在下列輸入:

from pandas import * 

data = read_csv('testProp.csv') 

temp = data.groupby('name')['prop'].describe() 
temp.to_csv('out.csv') 

輸出爲:

name  
A  count 3.000000 
     mean  2.000000 
     std  1.000000 
     min  1.000000 
     25%  1.500000 
     50%  2.000000 
     75%  2.500000 
     max  3.000000 
B  count 3.000000 
     mean  3.666667 
     std  1.527525 
     min  2.000000 
     25%  3.000000 
     50%  4.000000 
     75%  4.500000 
     max  5.000000 
dtype: float64 

然而,我想在格式的數據下面。我曾嘗試transpose()並想使用describe()維護和操作這些替代a .agg([np.mean(), np.max(), etc....)

count mean std min 25% 50% 75% max 
A 3 2 1 1 1.5 2 2.5 3 
B 3 3.666666667 1.527525232 2 3 4 4.5 5 

回答

2

一種方式做,這將是第一次做.reset_index(),重置索引您temp數據框中,然後用你可以隨意選擇DataFrame.pivot。示例 -

In [24]: df = pd.read_csv(io.StringIO("""name,prop 
    ....: A,1 
    ....: A,2 
    ....: B, 4 
    ....: A, 3 
    ....: B, 5 
    ....: B, 2""")) 

In [25]: temp = df.groupby('name')['prop'].describe().reset_index() 

In [26]: newdf = temp.pivot(index='name',columns='level_1',values=0) 

In [27]: newdf.columns.name = '' #This is needed so that the name of the columns is not `'level_1'` . 

In [28]: newdf 
Out[28]: 
     25% 50% 75% count max  mean min  std 
name 
A  1.5 2 2.5  3 3 2.000000 1 1.000000 
B  3.0 4 4.5  3 5 3.666667 2 1.527525 

然後,您可以將此newdf保存爲csv。

+0

不工作。給出「KeyError:0」 – Shehryar

0

在pandas v0.22中,您可以使用拆散功能。從上面的@Kumar回答開始,您可以使用熊貓堆棧/拆箱功能並播放它的變體。

from io import StringIO 
import pandas as pd 
df = pd.read_csv(StringIO("""name,prop 
    A,1 
    A,2 
    B, 4 
    A, 3 
    B, 5 
    B, 2""")) 

df.shape 
df 
temp = df.groupby(['name'])['prop'].describe() 
temp 
temp.stack() #unstack(),unstack(level=-1) level can be -1, 0 

退房文檔pandas unstack更多細節