如何在Rcpp中访问DataFrame的特定元素？

否，Rcpp::DataFrame没有像NumericMatrix那样直接的X(i,j)元素访问操作符。要访问DataFrame的元素，通常需要先提取对应的列（列会转为Vector类型），再按索引访问元素。

示例代码 #

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
int accessDataFrameElement(DataFrame df) {
    // 通过索引提取第一列（假设为整数类型）
    IntegerVector col1 = df[0];
    // 访问该列的第2个元素（Rcpp中索引从0开始）
    int elem = col1[1];
    
    // 也可以通过列名提取列
    IntegerVector col2 = df["target_col"];
    int elem2 = col2[0];
    
    return elem;
}

如果需要按行索引获取整行元素，可以遍历所有列并提取对应位置的值：

// [[Rcpp::export]]
List getRowElements(DataFrame df, int row_idx) {
    List row_data;
    // 遍历DataFrame的所有列
    for (int i = 0; i < df.size(); ++i) {
        Vector col = df[i];
        row_data.push_back(col[row_idx]);
    }
    // 保留原列名
    row_data.names() = df.names();
    return row_data;
}

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