我有一个数据集的日期,回报,投资组合和市值上限的股票列表。我想在我的数据和每个投资组合中计算每月价值加权的市场回报。
date ret portf mkval
1982-03-31 0.02 3.0 2000
1982-04-30 0.05 2.0 500
1982-05-31 0.10 1.0 3000
1982-03-31 0.05 3.0 4000
1982-04-30 0.20 3.0 700
1982-05-31 0.02 2.0 2000
1982-05-31 0.08 1.0 5000这些数据应产生以下输出:
date portf equal_w_ret
1982-03-31 3.0 0.04
1982-04-30 2.0 0.05
1982-04-30 3.0 0.20
1982-05-31 1.0 0.0875
1982-05-31 2.0 0.02在这里,第一行计算为:(2000/(2000+4000))(1+0.02)+(4000/(2000+4000))(1+0.05)-1
提前感谢!
发布于 2021-02-09 10:33:48
首先设置数据
data = { 'date' : ['1982-03-31','1982-04-30','1982-05-31','1982-03-31','1982-04-30','1982-05-31','1982-05-31'],
'ret' : [0.02,0.05,0.10,0.05,0.20,0.02,0.08],
'portf' : [3.0,2.0,1.0,3.0,3.0,2.0,1.0],
'mkval' : [2000,500,3000,4000,700,2000,5000]}现在,将数据生成数据并准备输出数据。
df = pd.DataFrame(data)
dfout = pd.DataFrame()有趣的一点!Groupby日期和项目组合号,然后按行计算。然后创建一个数据文件,这是该行的摘要,并将其放入输出数据。
for group, subdf in df.groupby(['date','portf']):
subdf['wret'] = (subdf['mkval'] * ( 1 + subdf['ret']))/subdf['mkval'].sum()
df2 = pd.DataFrame({ 'data' : [group[0]],'portf':[group[1]],'equal_w_ret':[subdf['wret'].sum() - 1]})
dfout = dfout.append(df2)这使得
data portf equal_w_ret
0 1982-03-31 3.0 0.0400
0 1982-04-30 2.0 0.0500
0 1982-04-30 3.0 0.2000
0 1982-05-31 1.0 0.0875
0 1982-05-31 2.0 0.0200https://stackoverflow.com/questions/66116727
复制相似问题