我要做的是找到多个大型矩阵之间的匹配日期。而我希望我的C ++代码返回的是行索引,其中找到了匹配项
我是C ++的新手,我发现加速我的R代码非常有用。
我的代码似乎在R Studio中工作但是在Rcpp函数被我的R代码中的另一个循环中的一些迭代使用后崩溃
以下是一些示例数据
baseflow_mat [[1]]是一个格式为
的矩阵baseflow_mat [[2]]是我想要找到匹配项的示例
baseflow_mat<-list()
baseflow_mat[[1]]<-data.frame(year=c(1992,1992,1992,1992),month=c(7,7,7,7),day=c(5,10,13,17),flow=c(50,60,59,33))
baseflow_mat[[2]]<-data.frame(year=c(1992,1992,1992,1992,1992,1992,1992,1992),month=c(7,7,7,7,7,7,8,8),day=c(4,10,13,18,26,27,2,6),flow=c(50,60,59,33,45,40,55,52))
我希望在baseflow_mat的所有170个大型矩阵中找到匹配的日期
baseflow_mat [[2]]是我想要找到匹配项的示例
所以我希望我的C ++代码返回的是baseflow_mat_1所有匹配的baseflow_mat_2的行索引。这适用于小矩阵,但一旦我开始使用我的所有数据,它就会崩溃。我的实际数据在每个矩阵中有500到3000行,我想找到170个单独矩阵的匹配,所以170 * 170约28900个结果
这是我的R代码
library(Rcpp)
sourceCpp("Source1.cpp")
big_match<-list()
for(i in seq(1,2)){#length(baseflow_mat))){
match_baseflow_list<-list()
for(j in seq(1,2)){#length(baseflow_mat))){
matches_wzeros<-matchRows(nrow(baseflow_mat[[j]][,1:3]),nrow(baseflow_mat[[i]][,1:3]),baseflow_mat[[j]][,1:3],baseflow_mat[[i]][,1:3])
matches<-matches_wzeros[matches_wzeros>0]
match_baseflow_list[[j]]<-baseflow_mat[[j]][matches,]
}
big_match[[i]]<-match_baseflow_list
}
这是我的C ++代码
// [[Rcpp::export]]
NumericVector matchRows(int rowSize, int matchRowSize, DataFrame nonMatchDF, DataFrame matchDF)
{
//0 is for year, 1 is for month, 2 is for day for both DF
Rcpp::NumericVector nonmatchYear = nonMatchDF[0];
Rcpp::NumericVector nonmatchMonth = nonMatchDF[1];
Rcpp::NumericVector nonmatchDay = nonMatchDF[2];
Rcpp::NumericVector matchYear = matchDF[0];
Rcpp::NumericVector matchMonth = matchDF[1];
Rcpp::NumericVector matchDay = matchDF[2];
Rcpp::NumericVector indexMatrix(matchRowSize*rowSize);
//j is for going through the nonmatch dataframe
int j;
//i is for going through the other DF
int i;
//addIndex is to add correctly to the vector
int addIndex = 0;
for (i = 0; i < matchRowSize; i++)
{
//Rcpp::NumericVector indexMatch(rowSize);
for (j = 0; j < rowSize; j++)
{
if ((matchYear[i] == nonmatchYear[j]) && (matchMonth[i] == nonmatchMonth[j]) && (matchDay[i] == nonmatchDay[j]))
{
indexMatrix[addIndex + j] = j + 1 ;
//indexMatrix(i, j) = j+1;
}
}
addIndex = addIndex + (j+1);
}
return indexMatrix;
}
sessionInfo()输出
R version 3.1.1 (2014-07-10)
Platform: x86_64-w64-mingw32/x64 (64-bit)
locale:
[1] LC_COLLATE=English_United States.1252
[2] LC_CTYPE=English_United States.1252
[3] LC_MONETARY=English_United States.1252
[4] LC_NUMERIC=C
[5] LC_TIME=English_United States.1252
attached base packages:
[1] stats graphics grDevices utils
[5] datasets methods base
other attached packages:
[1] Rcpp_0.11.6
loaded via a namespace (and not attached):
[1] tools_3.1.1
答案 0 :(得分:4)
你在最里面的循环中所做的事实上是merge
。如果您需要更快,请使用dplyr::left_join
(merge
的快速替代方法)。
big_match<-list()
for(i in seq(1,2)){#length(baseflow_mat))){
match_baseflow_list<-list()
for(j in seq(1,2)){#length(baseflow_mat))){
match_baseflow_list[[j]] <- merge(baseflow_mat[[1]], baseflow_mat[[2]], by=c("year", "month", "day"))
}
big_match[[i]]<-match_baseflow_list
}