rcpp命名为数字矩阵

时间:2016-03-05 16:47:39

标签: c++ r matrix rcpp

使用Rcpp,我无法找到通过C ++代码中的列名来寻址矩阵的方法。对于各种Matrix函数来说似乎没有重载,这些函数可以让你按照R中的规定通过名称来寻址行或列。我使用的用例就是你有一个sql查询等值的表,其中每列都有名。

这是我提出的明显不完整且不理想的工作:

class NamedNumericMatrix {
public:
    NamedNumericMatrix(SEXP m)
    {
      M=NumericMatrix(m);
      List dimnames = M.attr("dimnames");
      vector<string> colnames = dimnames[1];
      for(int i = 0; i<colnames.size(); i++){
        map<string, int>::iterator it = colNameIndex.find(colnames[i]);
        if(it != colNameIndex.end()){
          throw std::invalid_argument("duplicate colname found");
        }
        colNameIndex[colnames[i]] = colNameIndex.size()-1;
      }
    }
    double GetValue(int row, string col){
      map<string, int>::iterator it = colNameIndex.find(col);
      if(it == colNameIndex.end()){
        throw std::invalid_argument("col name not found");
      }
      return M(row, it->second);
    }
  int nrow(){
    return M.nrow();
  }
  int ncol(){
    return M.ncol();
  }
private:
    NumericMatrix M;
    map<string, int> colNameIndex;
};

我的问题是,使用Rcpp有更简单的方法吗?

1 个答案:

答案 0 :(得分:2)

只需使用返回字符向量的colnames()

R> cppFunction("int showme(NumericMatrix M) { print(colnames(M)); return 0; }")
R> showme(matrix(1:9,3,dimnames=list(NULL, c("a1", "b2", "c3"))))
[1] "a1" "b2" "c3"
[1] 0
R> 

哦,抱歉,重新阅读:您的名字是 index 吗?我们主要将矩阵视为数值对象。您可以按名称索引ListDataFrame。对于矩阵,您可能已使用上述内容回答了您的问题。