R语言自定义截尾均值函数返回NaN问题排查求助

你的自定义截尾均值函数返回NaN的核心原因是索引方式错误，导致最终处理后的x_trimmed变成了空向量，而mean()对空向量计算会返回NaN。

具体错误点分析 #

当传入示例参数trim=0.1时：

1. 输入向量长度n=10，计算得n_to_trim=floor(0.1*10)=1，即两端各截去1个元素。
2. lo_idx=1，x_sorted[!lo_idx]中，!1会被R解析为单个FALSE，用这个逻辑值索引向量时，会直接返回空向量（因为FALSE表示不选取任何元素）。
3. 后续对空向量x_trimmed再次索引，结果依然是空，最终mean(空向量)返回NaN。

另外，即使第一步没得到空向量，hi_idx=(10-1):10=9:10，当你对长度为9的x_trimmed（假设第一步正确截去第一个元素）用!hi_idx索引时，hi_idx是数值9和10，!9和!10都是FALSE，同样会导致错误的索引结果。

修复后的函数实现 #

可以通过两种方式修正：

方法1：使用逻辑向量标记保留元素

trimmed_mean <- function(x, trim) {
    n <- length(x)
    n_to_trim <- floor(trim * n)
    x_sorted <- sort(x)
    # 创建全为TRUE的逻辑向量，标记要保留的元素
    keep <- rep(TRUE, n)
    # 标记要截去的两端元素为FALSE
    keep[1:n_to_trim] <- FALSE
    keep[(n - n_to_trim + 1):n] <- FALSE  # 修正索引范围，避免重复截取
    x_trimmed <- x_sorted[keep]
    return(mean(x_trimmed))
}

方法2：直接切片（更简洁高效）

既然已经排序，直接通过索引范围截取中间保留的部分即可：

trimmed_mean <- function(x, trim) {
    n <- length(x)
    n_to_trim <- floor(trim * n)
    x_sorted <- sort(x)
    # 截取从n_to_trim+1到n-n_to_trim的元素
    x_trimmed <- x_sorted[(n_to_trim + 1):(n - n_to_trim)]
    return(mean(x_trimmed))
}

测试验证 #

运行你的测试代码：

trimmed_mean(c(-20, 1, 3, 2, 2, 5, 20, 2, 3, 4), trim = .1)

两种实现都会返回预期的2.75。

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