Python 3: Как суммировать результаты в словаре

Question

Я пишу код, который просматривал бы банковские выписки и совокупные расходы/доход в ведра в зависимости от месяца, в который они попадают. Данные находятся в форме csv и читаются как строковые записи. В файле csv есть 5 столбцов: столбец 1 для даты, 4 - для расходов и 5 для дохода.

Код должен:

1. Посмотрите месяц в дате строки из CSV.
2. Добавить расходы/доход в соответствующие словари.

Я попал в ловушку, пытаясь преобразовать расходы (формат: «$ 0.00») в поплавок и суммировать их в словарях. Кто-нибудь может сказать мне, что я могу здесь сделать?

Вот код:

import numpy as np 
import csv 
import timestring as ts 


months= [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12] 
expenses = {1:0, 2:0, 3:0, 4:0, 5:0, 6:0, 7:0, 8:0, 9:0, 10:0, 11:0, 12:0} 
income = {1:0, 2:0, 3:0, 4:0, 5:0, 6:0, 7:0, 8:0, 9:0, 10:0, 11:0, 12:0} 
exp_cat = [] 
income_cat = [] 

files =['export.csv', 'export1.csv'] 

with open("budgetfile.csv","wt") as fw: 
    writer = csv.writer(fw) 
    for file in files: 
     with open(file) as csvfile: 
      records = csv.reader(csvfile, quoting=csv.QUOTE_NONE) 
      print("Processing file {}. \n" .format(file)) 
      header = next(records) 
      for row in records: 
       try: 
        expenses[ts.Date(row[0]).month] += float(row[4]) 
        income[ts.Date(row[0]).month] += float(row[5]) 
        break 
       except ValueError: 
        pass  

Answer 1

Попробуйте

import numpy as np 
import csv 
import timestring as ts 
import decimal 
from collections import defaultdict 


months= [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12] 
expenses = defaultdict(decimal.Decimal) 
income = defaultdict(decimal.Decimal) 
'''Optional for less imports 
expenses = {x:decimal.Decimal() for x in months} 
income = {x:decimal.Decimal() for x in months} 
''' 
exp_cat = [] 
income_cat = [] 

files =['export.csv', 'export1.csv'] 

with open("budgetfile.csv","wt") as fw: 
    writer = csv.writer(fw) 
    for file in files: 
     with open(file) as csvfile: 
      records = csv.reader(csvfile, quoting=csv.QUOTE_NONE) 
      print("Processing file {}. \n" .format(file)) 
      header = next(records) 
      for row in records: 
       try: 
        expenses[ts.Date(row[0]).month] += decimal.Decimal(row[4].replace('$', '')) 
        income[ts.Date(row[0]).month] += decimal.Decimal(row[5].replace('$', '')) 
        break 
       except ValueError: 
        pass