Julia计算短期股票交易加权平均成本返回DataFrame结果异常
问题原因
该异常和数值舍入无关,是代码存在两处基础逻辑错误导致:
- 状态变量初始化位置错误:你把记录累计持仓量
s、累计持仓价值t、当前加权成本u的三个变量写在了for循环内部,每次迭代开始时三个变量都会被重置为0,完全无法保留上一笔交易后的持仓状态,自然无法正确累计计算。 - 返回语句位置错误:
return vwavg被写在了for循环块内部,第一次迭代完成后就会直接终止函数返回结果,根本不会执行后续所有行的计算。
另外你在函数内直接修改传入的volume数组(即原DataFrame的qty列),Julia中数组是引用传递,这种写法会直接污染原始数据,属于需要规避的副作用。
修正方案
- 将三个累计状态变量移到
for循环外部初始化,保证每轮计算的状态可以跨迭代留存 - 将
return vwavg移到for循环外部,等所有行计算完成后再返回结果 - 新增独立变量存储累计持仓,不直接修改输入的原始数据,避免污染源数据
修正后的可运行代码如下:
using DataFrames # 测试数据集 df = DataFrame(qty = [3, 2, 2, -7, 4, 4, -3,-2, 4, 4, -2, -3], price = [100.0, 99.0, 101.0, 103.0, 95.0, 93.0, 90.0, 90.0, 93.0, 95.0, 93.0, 92.0]) # 加权平均成本计算函数 function vwacost(volume, unitprice) n = length(volume) value = Vector{Float64}(undef, n) vwavg = Vector{Float64}(undef, n) # 累计状态初始化:移到循环外部 cum_qty = 0.0 cum_value = 0.0 current_cost = 0.0 for i in 1:n if volume[i] > 0 # 买入逻辑:累计价值、持仓量累加,重新计算加权成本 cum_value += volume[i] * unitprice[i] cum_qty += volume[i] current_cost = cum_value / cum_qty value[i] = cum_value vwavg[i] = current_cost else # 卖出逻辑:成本不变,扣减对应持仓量和价值 cum_qty += volume[i] cum_value = current_cost * cum_qty value[i] = cum_value vwavg[i] = current_cost end end # 返回语句移到循环外部 return vwavg end # 计算结果,指定输出列名为vwap out = transform(df, [:qty, :price] => vwacost => :vwap)
运行后输出的结果符合加权平均成本的计算逻辑,和逐行打印调试的单步结果一致。
内容的提问来源于stack exchange,提问作者Najib Noer
相关产品推荐
相关产品推荐

