Python Pandas – Lọc các hàng từ DataFrame dựa trên tổng giá trị

Để lọc các hàng từ DataFrame dựa trên tổng giá trị, ta xét ví dụ điểm số của 3 học sinh. Mục tiêu là giữ lại các hàng mà tổng điểm của 3 học sinh ở môn đó lớn hơn 200.

Tạo DataFrame

import pandas as pd

dataFrame = pd.DataFrame({
    'Jacob_Marks': [95, 90, 70, 85, 88],
    'Ted_Marks': [60, 50, 65, 85, 70],
    'Jamie_Marks': [77, 76, 60, 45, 50]
})

Lọc hàng có tổng lớn hơn 200

dataFrame = dataFrame[dataFrame.sum(axis=1) > 200]

Tham số axis=1 tính tổng theo hàng (ngang).

Ví dụ hoàn chỉnh

import pandas as pd

dataFrame = pd.DataFrame({
    'Jacob_Marks': [95, 90, 70, 85, 88],
    'Ted_Marks': [60, 50, 65, 85, 70],
    'Jamie_Marks': [77, 76, 60, 45, 50]
})

print("DataFrame gốc...\n", dataFrame)

dataFrame = dataFrame[dataFrame.sum(axis=1) > 200]

print("DataFrame đã lọc...\n", dataFrame)

Kết quả

DataFrame gốc...
   Jacob_Marks  Jamie_Marks  Ted_Marks
0           95           77         60
1           90           76         50
2           70           60         65
3           85           45         85
4           88           50         70

DataFrame đã lọc...
   Jacob_Marks  Jamie_Marks  Ted_Marks
0           95           77         60
1           90           76         50
3           85           45         85
4           88           50         70

Hàng thứ 2 (chỉ số 2) bị loại vì tổng điểm = 70+60+65 = 195 < 200.