我正在尝试对我的交易进行投资组合分析,并尝试计算调整后的成本基础价格。我几乎尝试了一切,但似乎没有任何效果。我能够计算调整后的数量,但无法获得调整后的购买价格有人可以帮忙吗?
这是示例交易日志原始数据
import pandas as pd
import numpy as np
raw_data = {'Date': ['04-23-2020', '05-05-2020', '05-05-2020', '05-11-2020', '05-11-2020',
'05-12-2020', '05-12-2020', '05-27-2020', '06-03-2020', '06-03-2020',
'06-03-2020', '06-03-2020', '06-03-2020'],
'Type': ['Buy', 'Buy', 'Buy', 'Buy', 'Buy', 'Buy', 'Buy',
'Sell', 'Sell', 'Sell', 'Buy', 'Sell', 'Sell'],
'Symbol': ['TSE:AC', 'TSE:AC', 'TSE:HEXO', 'TSE:BPY.UN', 'TSE:BPY.UN',
'TSE:BPY.UN', 'TSE:AC', 'TSE:BPY.UN', 'TSE:AC', 'TSE:BPY.UN',
'TSE:AC', 'TSE:BPY.UN', 'TSE:HEXO'],
'Quantity': [75, 100, 1450, 200, 50, 80, 150, 100, 125, 100, 100, 50, 1450],
'Amount per unit': [18.04, 17.29, 0.73, 13.04, 13.06, 12.65, 15.9, 15.01,
18.05, 14.75, 15.8, 14.7, 1.07],
'Turnover': [1353, 1729, 1058.5, 2608, 653, 1012, 2385, 1501, 2256.25, 1475, 1580, 735, 1551.5],
}
df = pd.DataFrame (raw_data, columns = ['Date','Type','Symbol','Quantity','Amount per unit', 'Turnover']).sort_values(['Date','Symbol']).reset_index(drop = True)
我能够毫无问题地获得调整后的数量,但无法获得正确的调整后的每单位价格。这里的条件是,如果我出售股票,我的调整后的每单位价格不应改变,并保持与购买该股票时的最后调整价格相同。
#to calculate adjusted quantity. this works as expected
df['Adjusted Quantity'] = df.apply(lambda x: ((x.Type == "Buy") - (x.Type == "Sell")) * x['Quantity'], axis = 1)
df['Adjusted Quantity'] = df.groupby('Symbol')['Adjusted Quantity'].cumsum()
#section where I am having problem. Works good until I reach the row where sell was made
df['Adjusted Price Per Unit'] = df.apply(lambda x: ((x.Type == "Buy") - (x.Type == "Sell")) * x['Turnover'], axis = 1)
df['Adjusted Price Per Unit'] = df.groupby('Symbol')['Adjusted Price Per Unit'].cumsum().div(df['Adjusted Quantity'])
运行此代码将产生以下结果
例如:索引 7 处的行的调整后价格应为 12.948(与索引 6 处的行相同),而不是 12.052。另外,最后一行调整后的价格应该是 0.73(与索引 2 的行相同),因为我买卖相同数量的股票。
例 2:在指数 6 处,我以 12.65 的价格买入 80 股 BPY,这使我的平均价格降至 12.94,总共 330 股(250+80)。现在,我以 15.01(指数 7)卖出 100 股。我的代码将调整后的成本调整为 12.05。我需要调整后的成本为 12.94,而不是 12.05。简单来说,如果交易类型是卖出,则忽略调整价格。使用该特定股票的上次购买类型交易中的上次调整价格。
我的代码的最后两行不正确。您能帮我正确计算调整后的每单位价格吗?谢谢 :)