将std :: map转换为Rcpp :: List?

时间:2015-12-09 14:21:41

标签: c++ r rcpp

std::map< std::string, double >作为Rcpp::List对象返回的正确方法是什么?默认的wrap()方法导致命名向量返回到R。

1 个答案:

答案 0 :(得分:5)

将来,请在您的问题中包含您尝试过的内容。当其他人能够解释为什么你正在尝试的东西不起作用时,对你来说会更有益。此外,R list s(和Rcpp::List s)比std::map更通用,因此我不太确定您是如何尝试存储地图的。列表中的数据。据推测,地图的密钥对应于列表的名称;以及列表值的值?这是实现此目的的一种方法:

#include <Rcpp.h>

// [[Rcpp::export]]
Rcpp::List map_to_list(Rcpp::CharacterVector cv, Rcpp::NumericVector nv) {
  std::map<std::string, double> my_map;

  for (R_xlen_t i = 0; i < cv.size(); i++) {
    my_map.insert(std::make_pair(Rcpp::as<std::string>(cv[i]), nv[i]));
  }

  Rcpp::List my_list(my_map.size());
  Rcpp::CharacterVector list_names(my_map.size());
  std::map<std::string, double>::const_iterator lhs = my_map.begin(), rhs = my_map.end();

  for (R_xlen_t i = 0; lhs != rhs; ++lhs, i++) {
    my_list[i] = lhs->second;
    list_names[i] = lhs->first;
  }

  my_list.attr("names") = list_names;
  return my_list;
}

/*** R

.names <- letters[1:20]
.values <- rnorm(20)

(res <- map_to_list(.names, .values))
#$a
#[1] -0.8596328

#$b
#[1] 0.1300086

#$c
#[1] -0.1439214

#$d
#[1] -1.017546

## etc...

*/ 

Rcpp::List my_list(my_map.size());开始,我将返回对象初始化为std::map的大小;同样地,Rcpp::CharacterVector将用作返回对象的名称。使用map迭代器和辅助索引变量,将映射的值(iterator->second)分配给列表中的相应位置,同时分配其键(iterator->first)到临时名称向量。最后,将其分配给结果列表,并将其返回。