Блог

Я пытаюсь суммировать столбец в файле csv с помощью python. Вот пример данных csv;

 Date,Profit/Losses
Jan-2010,867884
Feb-2010,984655
Mar-2010,322013
Apr-2010,-69417
May-2010,310503
Jun-2010,522857
Jul-2010,1033096
Aug-2010,604885
Sep-2010,-216386
 

Я хочу суммировать столбец прибыли / убытков.
Я использую следующий код, но он возвращает 0. Где я мог ошибиться?

 import os
import csv

# Path to collect data from the csv file in the Resources folder
pybank_csv = os.path.join("resources", "budget_data.csv")

with open(pybank_csv, 'r') as csvfile:       
   csvreader = csv.reader(csvfile, delimiter=',')
   next(csvfile, None)    
   t = sum(float(row[1]) for row in csvreader)

   #print the results
   print(f"Total: {t}")
 

Самый простой способ — использовать библиотеку pandas.

Используется pip install pandas для установки pandas на вашем компьютере

и затем

 import pandas as pd
df = pd.read_csv('your_filename.csv')
sumcol = df['Profit/Losses'].sum()
print(sumcol)
 

Теперь сумма находится в объекте sumcol. Для дальнейшего использования, если ваша задача — работать с данными, представленными в файле csv, pandas — это благословение. Эта библиотека предоставляет вам тысячи различных типов операций, которые вы можете выполнять с вашими данными. Для получения дополнительной информации обратитесь к веб-сайту Pandas.

Если вы хотите использовать только пакет csv, вы можете прочитать csv как dict, а затем суммировать запись прибыли / убытка dict для каждой строки

 total = 0
with open('your_filename.csv', newline='') as csvfile:
    data = csv.DictReader(csvfile)
    for row in data:
        total = total   int(row['Profit/Losses'])
print(total)
 

Или, если вы хотите использовать reader вместо dict reader, вам нужно игнорировать первую строку. Что-то вроде этого

 total = 0
with open('your_filename.csv', newline='') as csvfile:
    data = csv.reader(csvfile)
    for row in data:
        if not str(row[1]).startswith('P'):
            total = total   int(row[1])
 print(total)