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

Python - Chỉ tính tổng các hàng cụ thể của Pandas DataFrame


Để chỉ tính tổng các hàng cụ thể, sử dụng phương thức loc(). Đề cập chỉ mục hàng bắt đầu và kết thúc bằng toán tử :. Sử dụng loc(), bạn cũng có thể đặt các cột cần bao gồm. Chúng ta có thể hiển thị kết quả trong một cột mới.

Trước hết, hãy tạo một DataFrame. Chúng ta có bản ghi Sản phẩm bao gồm Kho mở đầu và Kho đóng:

dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"], "Opening_Stock": [300, 700, 1200, 1500], "Closing_Stock": [200, 500, 1000, 900]})

Tổng của một số hàng, ví dụ 2 hàng đầu. Tên cột cũng được đề cập trong loc() là Opening_Stock và Closing_Stock. Chúng ta hiển thị kết quả trong cột mới "Sum_Result":

dataFrame['Sum_Result'] = dataFrame.loc[0:1, ["Opening_Stock", "Closing_Stock"]].sum(axis=1)

Ví dụ

Dưới đây là mã hoàn chỉnh:

import pandas as pd
dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"], "Opening_Stock": [300, 700, 1200, 1500], "Closing_Stock": [200, 500, 1000, 900]})
print("DataFrame...\n", dataFrame)
# tổng của một số hàng
# Tên cột cũng được đề cập trong loc() là Opening_Stock và Closing_Stock
# hiển thị kết quả trong cột mới Sum_Result
dataFrame['Sum_Result'] = dataFrame.loc[0:1, ["Opening_Stock", "Closing_Stock"]].sum(axis=1)
print("\nSumming some rows...\n", dataFrame)

Kết quả

Điều này sẽ tạo ra đầu ra sau:

DataFrame...
  Closing_Stock  Opening_Stock    Product
0           200            300     SmartTV
1           500            700  ChromeCast
2          1000           1200     Speaker
3           900           1500    Earphone
Summing some rows...
  Closing_Stock  Opening_Stock    Product  Sum_Result
0           200            300     SmartTV       500.0
1           500            700  ChromeCast      1200.0
2          1000           1200     Speaker         NaN
3           900           1500    Earphone         NaN