如何将元组列表转换为CSV文件

问题描述 投票:-1回答:1

是编程的新手,并试图创建一个摊销表。在这里找到了一些很棒的问题和答案,但是现在我被困在尝试将结果转换为csv文件。

from datetime import date
from collections import OrderedDict
from dateutil.relativedelta import *
import csv

def amortization_schedule(rate, principal, period):
start_date=date.today()
#defining the monthly payment for a loan
payment = -float(principal / ((((1 + (rate / 12)) ** period) - 1) / ((rate / 12) * (1 + (rate / 12)) ** period)))

beg_balance = principal
end_balance = principal
period = 1


while end_balance > 0 and period <= 60 * 12:

    #Recalculate the interest based on the current balance
    interest_paid = round((rate / 12) * beg_balance, 2)

    #Determine payment based on whether or not this period will pay off the loan
    payment = round(min(payment, beg_balance + interest_paid), 2)
    principal = round(-payment - interest_paid, 2)


    yield OrderedDict([('Month', start_date),
                       ('Period', period),
                       ('Begin Balance', beg_balance),
                       ('Payment', payment),
                       ('Principal', principal),
                       ('Interest', interest_paid),
                       ('End Balance', end_balance)])


    #increment the counter, date and balance
    period +=1
    start_date += relativedelta(months=1)
    beg_balance = end_balance     

我尝试将this link用作解决方案的一部分,但最终得到的csv如下所示:

M,o,n,t,h
P,e,r,i,o,d
B,e,g,i,n, ,B,a,l,a,n,c,e
P,a,y,m,e,n,t
P,r,i,n,c,i,p,a,l
I,n,t,e,r,e,s,t
E,n,d, ,B,a,l,a,n,c,e

这是我转换为csv的代码。

for start_date, period, beg_balance, payment, principal, 
  interest_paid, end_balance in amortization_schedule(user_rate, 
  user_principal, user_period):
  start_dates.append(start_date)
  periods.append(period)
  beg_balances.append(beg_balance)
  payments.append(payment)
  principals.append(principal)
  interest_paids.append(interest_paid)
  end_balances.append(end_balance)
with open('amortization.csv', 'w') as outfile: 
  csvwriter = csv.writer(outfile)
  csvwriter.writerow(start_dates)
  csvwriter.writerow(periods)
  csvwriter.writerow(beg_balances)
  csvwriter.writerow(payments)
  csvwriter.writerow(principals)
  csvwriter.writerow(interest_paids)
  csvwriter.writerow(end_balances)

任何帮助将不胜感激!

python-3.x csv generator export-to-csv ordereddictionary
1个回答
0
投票
with open('amortization.csv', 'w', newline='') as outfile:
    fieldnames = ['Month', 'Period', 'Begin Balance', 'Payment',
                  'Principal', 'Interest', 'End Balance']

    csvwriter = csv.DictWriter(outfile, fieldnames)

    for line in amortization_schedule(user_rate, user_principal, user_period):
        csvwriter.writerow(line)

csv文件的写入代码。

collections.OrderedDict是字典,因此可能需要使用csv.DictWriter写字典。这是一本字典,所以您不需要所有您要转换为csv的行。

© www.soinside.com 2019 - 2024. All rights reserved.