Thực hiện chức năng trên cặp hàng trong Pandas dataframe

Question

Nói rằng tôi có dataframe sau:

>>> df=pd.DataFrame(data=['A','B','C','D','E'], columns=['Name']) 
>>> df 
    Name 
0 A 
1 B 
2 C 
3 D 
4 E 
>>> 

Tôi muốn tạo ra một danh sách các giá trị cho các hàng liền kề trong dataframe. Nếu tôi tạo chỉ mục các cặp, tôi có thể nhận kết quả đó bằng cách sử dụng groupby:

>>> df.index=[0,0,1,1,2] 
>>> df.groupby(level=0).agg(lambda x: list(x)) 
    Name 
0 [A, B] 
1 [C, D] 
2  [E] 

Cách hiệu quả nhất để làm điều này là gì?

Answer 1

Bạn có thể groupby bởi "kề" trong một đi (không có đột biến các DataFrame):

In [11]: g = df.groupby(df.index // 2) 

và sau đó làm bất cứ điều gì bạn cần làm:

In [12]: g.get_group(0) 
Out[12]: 
    Name 
0 A 
1 B 

In [13]: g.sum() 
Out[13]: 
    Name 
0 AB 
1 CD 
2 E