R中自定义函数嵌入prod()在geom_function绘图时返回常量的原因排查

核心背景：geom_function的传参规则 #

geom_function不会逐个遍历x轴的每个取值单独调用传入的函数，而是一次性生成完整的连续x值向量，整体传入函数的对应参数，要求函数返回和输入向量长度完全相同的输出向量，一一对应每个x值的计算结果。

带prod()的版本异常原因 #

produit1的异常完全来自循环内的赋值逻辑：

• 函数预分配了长度为20（和index向量长度一致）的probs向量，每个设计为存储单个标量结果。
• 当geom_function传入长度为N（比如101个连续采样点）的power向量时，每次循环执行的probs[i] <- vec[index[i]] * power会生成一个长度为N的临时向量。
• 此时要把长度为N的向量赋值给probs的单个存储位，R会自动仅保留临时向量的第一个元素存入对应位置，剩余N-1个元素直接丢弃，同时触发“替换长度不匹配”的运行警告。
• 循环结束后，probs的所有元素都是基于power向量的第一个值计算的固定结果，最后调用prod(probs)得到的是和x轴取值完全无关的常量，因此绘制出水平直线。
  单独调用函数时传入的power是单个标量，不会触发上述长度不匹配问题，因此运行正常。

逐次乘积累加版本正常的原因 #

produit2的计算逻辑天然适配向量传参场景：

• 初始值probs为标量1，第一次循环时和长度为N的power向量相乘后，直接生成长度为N的结果向量。
• 后续循环中，始终保持长度为N的向量做元素一一对应的乘法，最终返回的结果和输入power向量长度完全一致，每个位置对应单个x值的计算结果，完全符合geom_function的输出要求，因此曲线和手动循环计算的结果完全重合。

内容的提问来源于stack exchange，提问作者Guillaume Mulier