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

Python - Chỉ tính tổng các hàng cụ thể của Khung dữ liệu gấu trúc

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

Đầu tiên, chúng ta hãy tạo một DataFrame. Chúng tôi có hồ sơ Sản phẩm trong đó, bao gồm cả Kho mở và Kết thúc -

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, tức là 1 st hai hàng. Tên cột cũng được đề cập trong loc () tức là Opening_Stock và Closing_Stock. Chúng tôi đang 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ụ

Sau đâ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

# sum of some rows
# Column names also mentioned in the loc() i.e. Opening_Stock and Closing_Stock
# displaying result in a new column Sum_Result
dataFrame['Sum_Result'] = dataFrame.loc[0 : 1,["Opening_Stock" , "Closing_Stock"]].sum(axis = 1)
print"\nSumming some rows...\n",dataFrame

Đầu ra

Điều này sẽ tạo ra kết quả 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