在Kdb+/Q中对指定透视表实现线性插值的技术咨询

Hey there! 看你描述了这么一个结构清晰的Kdb+透视表，想要实现线性插值对吧？咱们先把数据结构再明确下，然后分两种常见的插值场景给你讲具体实现方式~

你的透视表结构与样例数据 #

场景1：按到期年份（Matureyear）插值 #

这种场景是补全某个分组下（比如State=CA, Callable=Y, Couponrate=3.0）1-30年的收益率数据，针对缺失年份用线性插值填充。

首先我先写一个通用的线性插值函数，然后基于这个函数处理目标分组：

// 通用线性插值函数：已知x轴点、对应y值，计算目标x0的插值结果
linearInterp:{[x;y;x0]
  // 确保x轴数据是升序的，避免binr函数出错
  if[not asc x; x:asc x; y:y asc x];
  // 找到x0所在的区间索引
  idx:binr[x;x0];
  // 处理边界情况：x0超出已知范围时取边界值（可根据需求调整）
  idx:idx where idx>=0; idx:idx where idx<count x;
  if[count idx=0; :first y];
  if[idx=count x-1; :last y];
  // 计算线性插值
  x1:x[idx]; x2:x[idx+1]; y1:y[idx]; y2:y[idx+1];
  y1 + (y2-y1)*(x0-x1)/(x2-x1)
}

// 1. 定义要插值的目标年份（1-30年）
targetYears:1+til 30;

// 2. 筛选目标分组的原始数据
groupData:select Matureyear,AAA,AA+,AA,AA- from t where State=`CA, Callable=`Y, Couponrate=3.0;

// 3. 对每个评级列进行插值，生成完整的年份数据
interpolatedData:([] Matureyear:targetYears);
interpolatedData:update 
  AAA:linearInterp[groupData[`Matureyear]; groupData[`AAA]; each targetYears],
  AA+:linearInterp[groupData[`Matureyear]; groupData[`AA+]; each targetYears],
  AA:linearInterp[groupData[`Matureyear]; groupData[`AA]; each targetYears],
  AA-:linearInterp[groupData[`Matureyear]; groupData[`AA-]; each targetYears]
from interpolatedData;

// 4. 补充分组维度信息
interpolatedData:update State:`CA, Callable:`Y, Couponrate:3.0 from interpolatedData;

场景2：按评级（Rating）插值 #

如果需要计算介于现有评级之间的收益率（比如AA+和AA之间的中间评级），我们可以先给评级分配连续权重，再用线性插值计算目标评级的收益率：

// 给现有评级分配连续权重（可根据实际评级体系调整）
ratingWeights:`AAA`AA+`AA`AA-!1 2 3 4;

// 基于权重的评级插值函数
ratingInterp:{[weights; yields; targetWeight]
  sortedWeights:asc keys weights;
  sortedVals:weights sortedWeights;
  sortedYields:yields sortedWeights;
  linearInterp[sortedVals; sortedYields; targetWeight]
}

// 示例：计算CA/Y/3.0/10年分组下，权重为2.5（介于AA+和AA之间）的收益率
targetRow:first select AAA,AA+,AA,AA- from t where State=`CA, Callable=`Y, Couponrate=3.0, Matureyear=10;
targetYield:ratingInterp[ratingWeights; targetRow; 2.5];
// 输出结果：3.8（介于AA+的3.7和AA的3.9之间）
targetYield

额外技巧：宽表转窄表简化批量插值 #

如果需要批量处理多个分组的插值，可以先把宽表转成窄表，分组处理后再转回宽表，操作会更灵活：

// 宽表转窄表：把评级列转成行维度
narrowT:ungroup update Ratings:flip(`AAA`AA+`AA`AA-!flip(AAA;AA+;AA;AA-)) from t;
narrowT:rename `Yield`Rating from `Ratings`val in narrowT;

// 现在narrowT的结构是：State,Callable,Couponrate,Matureyear,Rating,Yield
// 可以方便地按State,Callable,Couponrate,Rating分组，对Matureyear批量插值

备注：内容来源于stack exchange，提问作者ClaudeTi