循环中的Python偏移量

Question

我有一段代码来计算贷款的摊销情况，并允许违约（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()

Answer 1

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

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