如何在Rcpp中以更优雅的方式（避免双层循环）实现列表内各向量的求和？

当然可以！你完全不用写双层循环就能实现这个需求，而且代码会更简洁优雅，还能借助Rcpp和STL的优化函数保证性能。下面给你几种实用的实现方式：

方法一：用Rcpp内置的sum函数（最贴近R风格） #

Rcpp已经为数值向量封装了sum函数，直接调用就能替代手动写的内层求和循环，再配合C++11的范围for循环简化外层遍历，代码非常简洁：

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
List t_list(List r_list) {
  // 预先分配结果列表的空间，比逐个push_back更高效
  List results(r_list.size());
  int idx = 0;
  // 范围for循环遍历输入列表的每个元素
  for (auto& elem : r_list) {
    // 把SEXP转换为NumericVector后直接调用sum
    results[idx++] = sum(as<NumericVector>(elem));
  }
  return results;
}

方法二：用STL的std::accumulate函数 #

如果你更习惯用C++标准库的工具，可以用std::accumulate来做向量累加，需要额外包含<numeric>头文件：

#include <Rcpp.h>
#include <numeric> // 引入std::accumulate的头文件
using namespace Rcpp;

// [[Rcpp::export]]
List t_list(List r_list) {
  List results(r_list.size());
  int idx = 0;
  for (auto& vec : r_list) {
    NumericVector nv = as<NumericVector>(vec);
    // 用std::accumulate完成累加，0.0作为初始值避免整数溢出
    results[idx++] = std::accumulate(nv.begin(), nv.end(), 0.0);
  }
  return results;
}

方法三：用STL的std::transform实现函数式风格 #

如果你喜欢函数式编程的写法，可以用std::transform把输入列表的每个元素直接转换为对应的和，完全不用手动管理索引：

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
List t_list(List r_list) {
  List results(r_list.size());
  // 用transform遍历输入列表，将每个元素转换为求和结果后存入results
  std::transform(r_list.begin(), r_list.end(), results.begin(),
                 [](const SEXP& elem) {
                   return sum(as<NumericVector>(elem));
                 });
  return results;
}

测试效果 #

把上面任意一段代码用sourceCpp加载后，运行你的测试代码：

List <- list(c(1:3), c(4:6), c(7:9))
t_list(List)

会得到和你原来代码完全一致的输出：

[[1]]
[1] 6

[[2]]
[1] 15

[[3]]
[1] 24

为什么这些方法更好？ #

• 避免了手动写双层循环，减少了出错的概率，代码可读性更强
• 预先分配结果列表空间，比push_back的动态扩容更高效
• 用到的sum、std::accumulate、std::transform都是经过优化的函数，性能不会比手动循环差

内容来源于stack exchange