在地图键的矢量中查找多个字符串

Lookup multiple strings in a vector in a map's keys

本文关键字:查找 字符串 地图      更新时间:2023-10-16

我有一个键和值的列表。还有另一个向量具有一些键。我想在列表中搜索这些键,并使用相应的值进行进一步计算。

#Inside R environment
simil<-list("key"=c("A","B","C","D"),"val"=c(1.2,3.2,2.0,1.9)) 
someVec = c("B","D")
//CPP code
#include <RcppEigen>
#include <map>
using namespace Rcpp;
// [[Rcpp::export]]
NumericVector someFunc(GenericVector simil, CharacterVector someVec){
List sim(simil);
std::map<std::string, double> mymap;
std::map<std::string, double>::iterator it;
CharacterVector keys = sim["key"];
NumericVector vals = sim["val"];
map[keys] = vals;
for(int i=0;i<someVec.size();i++){
it = mymap.find(someVec(i));
if(it!=mymap.end()){
// some logic here
}
} 

我不确定map[keys] = vals是否可以工作,因为这看起来更像是 R 语句,但编译器不会在此行中抛出任何错误。问题在于在地图上使用 .find()。 它抛出一个错误:没有匹配的成员函数供调用查找。

如何在该地图中搜索键?甚至mymap.count(someVec(i))>0也显示错误。

我不确定map[keys] = vals是否可以工作,因为这看起来更像是R语句,但编译器不会在此行中抛出任何错误。

我不确定为什么会编译,但它绝对不会做你期望它做的事情;std::map是一个C++类,不遵守 R 的矢量化语义。它有几个不同的构造函数,可用于用数据填充它,但在你的情况下,最简单的方法可能是循环你的输入并迭代填充它。

问题在于在地图上使用 .find()。它抛出一个错误:没有匹配的成员函数供调用查找。

我怀疑这与CharacterVector::operator[]返回string_proxy(或const_string_proxy,取决于上下文)的事实有关。这不会隐式转换为std::string,但您可以使用Rcpp::as<std::string>执行显式转换。

一起

#include <Rcpp.h>
#include <map>
using namespace Rcpp;
// [[Rcpp::export]]
NumericVector find_keys(List keyvals, CharacterVector morekeys) {
std::map<std::string, double> mymap;
std::map<std::string, double>::iterator it;
CharacterVector keys = keyvals["key"];
NumericVector vals = keyvals["val"];
R_xlen_t i = 0, n = keys.size(), m = morekeys.size();
NumericVector res(m, NA_REAL);
// fill map
for (; i < n; i++) {
mymap[as<std::string>(keys[i])] = vals[i];
}
// look up potential keys
for (i = 0; i < m; i++) {
it = mymap.find(as<std::string>(morekeys[i]));
if (it != mymap.end()) {
res[i] = it->second;
}
}
return res;
}
/*** R
keyvals <-list(
key = c("A","B","C","D"),
val = c(1.2,3.2,2.0,1.9)
)
morekeys = c("B", "D", "x")
find_keys(keyvals, morekeys)
# [1] 3.2 1.9  NA
*/

或者,Rcppwrap/as专业化知道如何"开箱即用"地处理CharacterVectorstd::vector<std::string>之间的转换,即

// [[Rcpp::export]]
NumericVector find_keys2(List keyvals, const std::vector<std::string>& morekeys) {
// ...
std::vector<std::string> keys = as<std::vector<std::string> >(keyvals["key"]);
// ...
// fill map
for (; i < n; i++) {
mymap[keys[i]] = vals[i];
}
// look up potential keys
for (i = 0; i < m; i++) {
it = mymap.find(morekeys[i]);
// ...
}
// ...
}