Pandas python .describe() định dạng/đầu ra

Question

Tôi đang cố gắng để có được hàm .describe() để xuất theo cách được định dạng lại. Đây là dữ liệu csv (testProp.csv)

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

khi tôi gõ như sau:

from pandas import * 

data = read_csv('testProp.csv') 

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

đầu ra là:

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 

Tuy nhiên, tôi muốn dữ liệu trong định dạng phía dưới. Tôi đã cố gắng transpose() và muốn duy trì sử dụng describe() và thao tác thay vì 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 

Answer 1

Một cách để làm điều này là lần đầu tiên làm .reset_index(), để đặt lại chỉ mục cho temp DataFrame của bạn, và sau đó sử dụng DataFrame.pivot như bạn muốn. Ví dụ -

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 

Sau đó, bạn có thể lưu số này newdf vào csv.

Answer 2

Trong gấu trúc v0.22, bạn có thể sử dụng tính năng unstack. Xây dựng từ câu trả lời @Kumar ở trên, bạn có thể sử dụng tính năng ngăn xếp/ngăn xếp gấu trúc và chơi xung quanh với biến thể của nó.

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 

Kiểm tra các tài liệu hướng dẫn pandas unstack để biết thêm chi tiết