从Rcpp函数返回NumericVectors列表

时间:2017-02-21 10:37:17

标签: r rcpp

我正在尝试计算Gower在一组项目之间的相似性。使用Rcpp包我正在编写自己的函数来计算相似度值,因为雏菊函数会抛出错误。

功能是:

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
List gowerSim(CharacterMatrix inp) {

  int n_row = inp.nrow(), n_col = inp.ncol();
  int sumRow = 0, colLen;
  List out(n_row);

  //double sim[n_row];
  NumericVector sim(n_row);

  for (int i = 0; i < n_row; i++) {

    for (int j = 0; j < n_row; j++) {

      sumRow = 0; 
      colLen = n_col;

      for (int k = 0; k < n_col; k++) {
        if (inp(i,k) != "NA" && inp(j,k) != "NA") {

          if (inp(i,k) != inp(j,k)) {
            sumRow = sumRow + 1;
          }
        } else {
          colLen = colLen - 1;
        }
      }
      if (colLen > 0) {
        sim[j] = (double) sumRow/colLen;
        //printf("%f",sim[j]);
      } else {
        sim[j] = NA_INTEGER;
      }
    }

    out[i] = sim;

    if (i < 3) {
      print(out);
    }
  }

  return out;
}

/*** R
clust<-gowerSim(inp)
*/

返回的列表将最后一个向量复制到所有其他元素,即假设clust的长度为250,clust[[1]]clust[[250]]的所有值都相同。但是,在打印(对于前三个元素)时,每个向量out[1]out[2]out[3]都不同。

有人可以告诉我这里有什么问题吗?

1 个答案:

答案 0 :(得分:1)

此问题的解决方案是在第一个sim命令之后定义向量for,如下所示:

List gowerSim(CharacterMatrix inp) {

  int n_row = inp.nrow(), n_col = inp.ncol();
  int sumRow=0,colLen;
  List out(n_row);

  for(int i=0;i<n_row;i++){

    NumericVector sim(n_row);

    for(int j=0;j<n_row;j++){
      sumRow=0; 
      colLen=n_col;
      for(int k=0; k<n_col;k++){
        if(inp(i,k)!="NA" && inp(j,k)!="NA"){
          if(inp(i,k)!=inp(j,k)){
            sumRow=sumRow+1;
          }
        }else{
          colLen=colLen-1;
        }
      }
      if(colLen>0){
        sim[j] = (double) sumRow/colLen;
        //printf("%f",sim[j]);
      }else{
        sim[j] = NA_INTEGER;
      }
    }
    out[i] = sim;
    if(i<3){
      print(out);
    }
  }

  return out;
}

一个小例子:

mat <- matrix( as.character(c(rep(1,5),sample(3,15,repl=TRUE),rep(5,5))),5)
clust <- gowerSim(mat)
clust

enter image description here

或者你可以像你一样定义矢量并在第一个for循环中重置它。

为什么这种方法确实有效,而你却不知道:我真的不知道,但我认为它是指C ++中的列表结构。

我解决问题的第一种方法是以下方法:相反填写一个列表,我们填写一个矩阵,这很好用,请看这里:

NumericMatrix gowerSim(CharacterMatrix inp) {

  int n_row = inp.nrow(), n_col = inp.ncol();
  int sumRow=0,colLen;
  NumericMatrix out(n_row, n_col);
  NumericVector sim(n_row);

  for(int i=0;i<n_row;i++);

    for(int j=0;j<n_row;j++){
      sumRow=0; 
      colLen=n_col;
      for(int k=0; k<n_col;k++){
        if(inp(i,k)!="NA" && inp(j,k)!="NA"){
          if(inp(i,k)!=inp(j,k)){
            sumRow=sumRow+1;
          }
        }else{
          colLen=colLen-1;
        }
      }
      if(colLen>0){
        sim[j] = (double) sumRow/colLen;
        //printf("%f",sim[j]);
      }else{
        sim[j] = NA_INTEGER;
      }
    }
    out(_,i) = sim;
    if(i<3){
      print(out);
    }
  }

  return out;
}