Computer >> Máy Tính >  >> Lập trình >> Python

Từng nhóm và tính tổng trong Python Pandas

Để tìm từng nhóm và tính tổng trong Python Pandas, chúng ta có thể sử dụng groupby (cột) .sum () .

Các bước

  • Tạo dữ liệu dạng bảng hai chiều, có thể thay đổi kích thước, có khả năng không đồng nhất, df .
  • In DataFrame đầu vào, df .
  • Tìm tổng theo nhóm bằng cách sử dụng df.groupby (). sum () . Hàm này nhận một cột nhất định và sắp xếp các giá trị của nó. Sau đó, dựa trên các giá trị đã sắp xếp, nó cũng sắp xếp các giá trị của các cột khác.
  • In tổng theo nhóm.

Ví dụ

import pandas as pd

df = pd.DataFrame(
    {
       "Apple": [5, 2, 7, 0],
       "Banana": [4, 7, 5, 1],
       "Carrot": [9, 3, 5, 1]
    }
)

print "Input DataFrame 1 is:\n", df
g_sum = df.groupby(['Apple']).sum()
print "Group by Apple is:\n", g_sum

Đầu ra

Input DataFrame 1 is:

   Apple Banana Carrot
0    5     4      9
1    2     7      3
2    7     5      5
3    0     1      1

Group by Apple is:

Apple Banana Carrot
0       1      1
2       7      3
5       4      9
7       5      5