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

Giới thiệu

Để lọc giữ lại các cột mà tổng giá trị của cột đó lớn hơn một ngưỡng, dùng loc[:, df.sum() > threshold].

Ví dụ minh họa

import pandas as pd

# DataFrame điểm 3 học sinh (3 cột)
dataFrame = pd.DataFrame({
    'Jacob_Marks': [95, 90, 75, 85, 88],
    'Ted_Marks': [60, 50, 65, 85, 70],
    'Jamie_Marks': [77, 76, 65, 45, 50]
})
print("DataFrame gốc...\n", dataFrame)

# Lọc cột có tổng > 400
dataFrame = dataFrame.loc[:, dataFrame.sum(axis=0) > 400]
print("\nDataFrame sau lọc (tổng cột > 400):\n", dataFrame)

Kết quả đầu ra

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

DataFrame sau lọc (tổng cột > 400):
   Jacob_Marks
0           95
1           90
2           75
3           85
4           88

Giải thích: Tổng Jacob=433 (>400), Ted=330, Jamie=313. Chỉ giữ cột Jacob_Marks.