循环中的Python偏移量

时间:2017-12-16 10:34:09

标签: python pandas loops generator ordereddictionary

我有一段代码可以计算贷款的摊销情况,并允许违约(cdr =常数违约率)和预付款(cpr =固定预付款率)。

我还想包括恢复,但这种恢复应该在未来一段时间内收到。在下面的例子中,我申请了3%的cdr,我希望60%的违约贷款余额在6个月后被收回。 我正在努力解决这个问题,因为在每个循环中它都需要引用回到前一个时期。

通过首先创建没有恢复的表/数据帧来解决这个问题,然后在第二阶段通过在列恢复中添加60%的默认值来应用恢复,抵消6个月。

但是我希望在amortize函数中有更好/更清晰的方法。

任何帮助都将不胜感激。

import pandas as pd
import numpy as np
from datetime import date
from collections import OrderedDict
from dateutil.relativedelta import *
pd.options.display.float_format = '{:,.2f}'.format

def amortize(principal, int_rate,periods, cpr, cdr, date, recovery_rate, recovery_timing):
    p = 0
    beg_balance = principal
    end_balance = principal

    while end_balance > 1:
        default = round((1-(1-cdr/100)**(1/12)) * beg_balance,2)
        interest = round((int_rate/12)*max(beg_balance-default,0),2)        
        if p < periods:
            pmt = -round(np.pmt(int_rate/12, periods -p, 
                                beg_balance - default),2)
        else:
            pmt = 0
        principal = pmt - interest
        prepay = round((1-(1-cpr/100)**(1/12)) * (beg_balance - principal),2)

        end_balance = max(beg_balance - principal - prepay - default,0)
        recovery = default * recovery_rate/100

        total_cash = pmt + prepay + recovery #plus a recovery lag
        yield OrderedDict([('Period',p+1),
                          ('Month', date),
                          ('Begin_Bal', beg_balance),
                          ('Default',default),
                          ('Sched Princ',principal),
                          ('Prepay Princ',prepay),
                          ('Interest',interest),
                          ('Recovery',recovery),
                          ('Total CF',total_cash),
                          ('End Balance', end_balance)])
        p += 1
        date += relativedelta(months=1)
        beg_balance = end_balance


table = amortize(300000,0.03,360,10,3,date(2017,12,11),60,6)                            

pd.DataFrame(table).head()

1 个答案:

答案 0 :(得分:0)

你的解决方案听起来很干净您还可以使用defaults.append(..)创建过去默认列表,然后参阅defaults[-6]

如果贷款期限很长,您可以使用固定容量为6的collections.deque而不是列表,以避免过多的旧计算。