如何将计算元素差和的Python for循环函数翻译为等效R代码
Python函数转R语言实现方案
原R代码错误原因
- return语句放置在双层循环的最内层,第一次迭代(i和j均取列表第一个元素)时就会直接返回结果,未完成全部组合的计算
- 未设置
i != j的过滤条件,且每次循环都会直接覆盖new_list的值,没有收集所有符合条件的差值
正确实现代码
与Python逻辑完全对齐的循环版本
sum_differences <- function(a){ new_list <- c() for (i in a){ for (j in a){ if (i != j) { new_list <- c(new_list, abs(i - j)) } } } outcome <- sum(new_list) / length(a) return(round(outcome, 2)) }
R风格向量化优化版本(无需循环,执行效率更高)
sum_differences <- function(a) { all_pairs <- expand.grid(i = a, j = a) valid_diff <- abs(all_pairs$i - all_pairs$j)[all_pairs$i != all_pairs$j] round(sum(valid_diff) / length(a), 2) }
测试验证
输入测试值a <- c(5, 3, 1),调用函数后返回结果为5.33,与预期的16/3计算结果一致。
内容的提问来源于stack exchange,提问作者Theresa_S
相关产品推荐
相关产品推荐

