我有一段代码可以计算贷款的摊销情况,并允许违约(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()
答案 0 :(得分:0)
你的解决方案听起来很干净您还可以使用defaults.append(..)
创建过去默认列表,然后参阅defaults[-6]
如果贷款期限很长,您可以使用固定容量为6的collections.deque
而不是列表,以避免过多的旧计算。