Блог

Я начинаю изучать Python. Как мне вычислить сумму (с плавающей запятой) определенного столбца в файле csv?

Это то, что я делал до сих пор, но мой код не дает мне общее количество ‘amount’ (столбец):

 giftFile = open('input.v0.small.csv')
giftReader = csv.reader(giftFile)
giftData = list(giftReader)

for row in giftReader:
    if len(row)>0:
        giftData  = row['amount']

print('row 0:'   str(giftData[0]))
print("row 1's dollar value: "   str(giftData[1]))
 

Пример из csv-файла:

 date,amount,vendor,not sure,who,category
3/11/17, 100.00,"99 PLEDGCharlie S 99PLEDGES.COMAZ",2,Bob Smith,charity
3/11/17, 3.27,"CAFE BEARYUM ORO VALLEY AZ"
 

Вы могли бы попробовать:

 import csv
from math import fsum

with open('input.v0.small.csv', 'r') as file:
    result = fsum(
        float(d['amount']) if d['amount'].strip() else 0
        for d in csv.DictReader(file) if d['amount']
    )
 

• Предположение: csv-файл содержит строку заголовка, а столбец, который вы хотите суммировать, называется amount .
• Использование a DictReader вместо a reader для получения суммы по имени столбца.
• Используя fsum вместо sum , чтобы избежать потери точности.

Результат для файла

 date,amount,vendor,not sure,who,category
3/11/17, 100.00,"99 PLEDGCharlie S 99PLEDGES.COMAZ",2,Bob Smith,charity
3/11/17, 3.27,"CAFE BEARYUM ORO VALLEY AZ"
 

является

 103.27
 

Результат для файла

 date,amount,vendor,not sure,who,category
3/11/17, 100.00,"99 PLEDGCharlie S 99PLEDGES.COMAZ",2,Bob Smith,charity


3/11/17, 3.27,"CAFE BEARYUM ORO VALLEY AZ"
 

является

 103.27
 

Вы можете сделать это легко с sum помощью . csv.reader возвращает итерацию, дающую список для каждой строки, поэтому нам просто нужно выбрать правильный элемент столбца, преобразовать в число с плавающей запятой, а затем добавить их все. В этом примере используются Decimal «ожидаемые» результаты при суммировании чисел с плавающей запятой, но вы можете использовать float вместо этого, если хотите:

 import csv
from decimal import Decimal

col = 1

with open('input.v0.small.csv') as giftFile:
    reader = csv.reader(giftFile)
    header = next(reader)
    print(sum(Decimal(x[col]) for x in reader if x[col]))
 

Просто измените col номер столбца для обработки.

При открытии файлов в Python файлы должны быть закрыты. Чтобы не забыть закрыть, используйте диспетчер контекста. После чтения ваших данных в виде списка списков вы можете использовать понимание списка для извлечения данных из указанного столбца. Затем используйте sum встроенную функцию для выполнения окончательного суммирования.

 from csv import reader

# context manager for reading-in CSV file 
with open('input.v0.small.csv', 'r') as read_obj:
    csv_reader = reader(read_obj)
    data = list(csv_reader)

# list comprehension to extract specified column
ls = [float(row[1]) for row in data[1:]] 

# built-in sum function
column_sum = sum(ls)

# print result
print(f'Column sum is {column_sum}')