如何在RCpp中向dataframe添加新列？

#include <Rcpp.h>
#include <RcppArmadillo.h>

// [[Rcpp::export]]
Rcpp::DataFrame addColumnToDataFrame(Rcpp::NumericVector columnData, std::string columnName, Rcpp::DataFrame df) {
  // 将列数据转换为arma::vec类型
  arma::vec col(columnData.begin(), columnData.size(), false);
  
  // 将dataframe转换为arma::mat类型
  arma::mat mat = Rcpp::as<arma::mat>(df);
  
  // 在mat中添加新列
  mat.insert_cols(mat.n_cols, col);
  
  // 创建新的dataframe对象
  Rcpp::DataFrame newDf(mat);
  
  // 设置新的列名
  Rcpp::CharacterVector colNames = newDf.names();
  colNames[colNames.size()] = columnName;
  newDf.attr("names") = colNames;
  
  return newDf;
}

# 调用C++函数
newDf <- addColumnToDataFrame(columnData, columnName, df)