Pandas sắp xếp theo nhóm tổng hợp và cột

Question

Với dataframe sau

In [31]: rand = np.random.RandomState(1) 
     df = pd.DataFrame({'A': ['foo', 'bar', 'baz'] * 2, 
          'B': rand.randn(6), 
          'C': rand.rand(6) > .5}) 

In [32]: df 
Out[32]:  A   B  C 
     0 foo 1.624345 False 
     1 bar -0.611756 True 
     2 baz -0.528172 False 
     3 foo -1.072969 True 
     4 bar 0.865408 False 
     5 baz -2.301539 True 

Tôi muốn sắp xếp nó vào nhóm (A) bằng tổng tổng hợp của B, và sau đó bởi các giá trị trong C (không cộng dồn) . Vì vậy, về cơ bản có được thứ tự của các A nhóm với

In [28]: df.groupby('A').sum().sort('B') 
Out[28]:    B C 
     A    
     baz -2.829710 1 
     bar 0.253651 1 
     foo 0.551377 1 

Và sau đó bởi True/False, để nó cuối cùng trông như thế này:

In [30]: df.ix[[5, 2, 1, 4, 3, 0]] 
Out[30]: A   B  C 
    5 baz -2.301539 True 
    2 baz -0.528172 False 
    1 bar -0.611756 True 
    4 bar 0.865408 False 
    3 foo -1.072969 True 
    0 foo 1.624345 False 

Làm thế nào điều này có thể được thực hiện?

Answer 1

groupby Một :

In [0]: grp = df.groupby('A') 

Trong mỗi nhóm, cộng tổng B và truyền giá trị s bằng cách sử dụng biến đổi. Sau đó, sắp xếp theo B:

In [1]: grp[['B']].transform(sum).sort('B') 
Out[1]: 
      B 
2 -2.829710 
5 -2.829710 
1 0.253651 
4 0.253651 
0 0.551377 
3 0.551377 

Chỉ mục df ban đầu bằng cách chuyển chỉ mục từ trên xuống. Điều này sẽ sắp xếp lại các giá trị A bằng tổng tổng hợp của các giá trị B:

In [2]: sort1 = df.ix[grp[['B']].transform(sum).sort('B').index] 

In [3]: sort1 
Out[3]: 
    A   B  C 
2 baz -0.528172 False 
5 baz -2.301539 True 
1 bar -0.611756 True 
4 bar 0.865408 False 
0 foo 1.624345 False 
3 foo -1.072969 True 

Cuối cùng, sắp xếp các giá trị 'C' trong các nhóm của 'A' bằng cách sử dụng tùy chọn sort=False để giữ gìn trật tự Một loại từ bước 1:

In [4]: f = lambda x: x.sort('C', ascending=False) 

In [5]: sort2 = sort1.groupby('A', sort=False).apply(f) 

In [6]: sort2 
Out[6]: 
     A   B  C 
A 
baz 5 baz -2.301539 True 
    2 baz -0.528172 False 
bar 1 bar -0.611756 True 
    4 bar 0.865408 False 
foo 3 foo -1.072969 True 
    0 foo 1.624345 False 

sạch lên chỉ số df bằng cách sử dụng reset_index với drop=True:

In [7]: sort2.reset_index(0, drop=True) 
Out[7]: 
    A   B  C 
5 baz -2.301539 True 
2 baz -0.528172 False 
1 bar -0.611756 True 
4 bar 0.865408 False 
3 foo -1.072969 True 
0 foo 1.624345 False 

Answer 2

Một cách để làm điều này là để chèn một cột giả với số tiền để sắp xếp:

In [10]: sum_B_over_A = df.groupby('A').sum().B 

In [11]: sum_B_over_A 
Out[11]: 
A 
bar 0.253652 
baz -2.829711 
foo 0.551376 
Name: B 

in [12]: df['sum_B_over_A'] = df.A.apply(sum_B_over_A.get_value) 

In [13]: df 
Out[13]: 
    A   B  C sum_B_over_A 
0 foo 1.624345 False  0.551376 
1 bar -0.611756 True  0.253652 
2 baz -0.528172 False  -2.829711 
3 foo -1.072969 True  0.551376 
4 bar 0.865408 False  0.253652 
5 baz -2.301539 True  -2.829711 

In [14]: df.sort(['sum_B_over_A', 'A', 'B']) 
Out[14]: 
    A   B  C sum_B_over_A 
5 baz -2.301539 True  -2.829711 
2 baz -0.528172 False  -2.829711 
1 bar -0.611756 True  0.253652 
4 bar 0.865408 False  0.253652 
3 foo -1.072969 True  0.551376 
0 foo 1.624345 False  0.551376 

và có thể bạn sẽ thả hàng giả:

In [15]: df.sort(['sum_B_over_A', 'A', 'B']).drop('sum_B_over_A', axis=1) 
Out[15]: 
    A   B  C 
5 baz -2.301539 True 
2 baz -0.528172 False 
1 bar -0.611756 True 
4 bar 0.865408 False 
3 foo -1.072969 True 
0 foo 1.624345 False 

Answer 3

Dưới đây là một cách tiếp cận ngắn gọn hơn ...

df['a_bsum'] = df.groupby('A')['B'].transform(sum) 
df.sort(['a_bsum','C'], ascending=[True, False]).drop('a_bsum', axis=1) 

Dòng đầu tiên thêm cột vào khung dữ liệu với tổng số theo nhóm. Dòng thứ hai thực hiện sắp xếp và sau đó loại bỏ cột phụ.

Kết quả:

A  B   C 
5 baz  -2.301539 True 
2 baz  -0.528172 False 
1 bar  -0.611756 True 
4 bar  0.865408 False 
3 foo  -1.072969 True 
0 foo  1.624345 False 

LƯU Ý: sort bị phản đối, sử dụng sort_values thay