DataTable技术问题:删除求和为0的行及计算非0值行均值
解决DataTable数据透视表的两个需求:删除求和为0的行 + 计算排除0后的行均值
嘿,我来帮你搞定这两个问题!先梳理下你代码里的小错误,再给出完整的解决方案:
1. 删除所有求和为0的行
你之前的代码有两个关键问题:
- 用了
sum(x) = 0,这是赋值操作,应该用相等判断的sum(x) != 0(我们要保留求和不为0的行) apply函数的括号没闭合,语法不完整
另外要注意:如果你的数据透视表第一列是名称(比如示例里的行A/B/C),需要排除这一列再计算求和,避免字符列干扰数值计算。
正确代码如下:
# 假设dt是你的数据透视表,第一列为名称,后续列为年份数值 dt_filtered <- dt[apply(dt[, -1], 1, function(x) sum(x) != 0), ]
2. 计算每行排除0值后的均值
我们可以用apply遍历每一行,先筛选出非0的数值,再计算均值。为了避免极端情况(比如全0行,虽然已经过滤过,但加个判断更稳妥)导致报错,我们可以增加长度判断:
# 在过滤后的数据集上添加均值列 dt_filtered$mean_excl_zero <- apply(dt_filtered[, -1], 1, function(x) { # 提取当前行非0的数值 non_zero_vals <- x[x != 0] # 如果没有非0值返回NA,否则计算均值 if (length(non_zero_vals) == 0) NA else mean(non_zero_vals) })
完整示例(模拟数据)
为了让你更直观看到效果,这里用模拟数据演示整个流程:
# 模拟你的数据透视表 dt <- data.frame( Name = c("A", "B", "C"), 2020 = c(10, 0, 0), 2021 = c(20, 30, 0), 2022 = c(0, 15, 0) ) # 执行需求1:删除求和为0的行(行C会被移除) dt_filtered <- dt[apply(dt[, -1], 1, function(x) sum(x) != 0), ] # 执行需求2:添加排除0后的均值列 dt_filtered$mean_excl_zero <- apply(dt_filtered[, -1], 1, function(x) { non_zero_vals <- x[x != 0] mean(non_zero_vals) }) # 查看结果 print(dt_filtered)
运行后输出的结果:
Name 2020 2021 2022 mean_excl_zero 1 A 10 20 0 15.0 2 B 0 30 15 22.5
内容的提问来源于stack exchange,提问作者Peter
相关产品推荐
相关产品推荐

