Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Đọc CSV tính tổng 2 hàng cuối cùng Python pandas (3 cách)

Giả sử bạn có file pandas.csv và muốn tính tổng 2 hàng cuối cùng theo 3 cách khác nhau.

Dữ liệu CSV

Id,Data
1,11
2,22
3,33
4,44
5,55
6,66
7,77
8,88
9,99
10,100

Cách 1: tail(2).sum()

import pandas as pd
data = pd.read_csv('pandas.csv')
df = pd.DataFrame(data)
print("Tổng 2 hàng cuối (tail):", df.tail(2).sum())

Cách 2: iloc[-2:].sum()

print("Tổng 2 hàng cuối (iloc):", df.iloc[-2:].sum())

Cách 3: Vòng lặp thủ công

id_sum = 0
data_sum = 0
for i in range(len(df)):
    if i == len(df)-2 or i == len(df)-1:
        id_sum += df.iloc[i, 0]
        data_sum += df.iloc[i, 1]
print("Id:", id_sum, "Data:", data_sum)

Kết quả (cả 3 cách)

Id    19
Data  199

Giải thích:

  • Hàng cuối: Id=9, Data=99. Hàng lפני cuối: Id=10, Data=100.
  • Tổng Id = 19, Tổng Data = 199.
  • Cách 1 và 2 ngắn gọn, vector hóa (khuyên dùng). Cách 3 minh họa logic thủ công.