在R中定义一个矩阵并将其传递给C ++

时间:2012-11-10 18:06:36

标签: c++ r rcpp

我在R中定义了一个矩阵。我需要将这个矩阵传递给c ++函数并在C ++中进行操作。 示例:在R中,定义矩阵

A <- matrix(c(9,3,1,6),2,2,byrow=T)
PROTECT( A = AS_NUMERIC(A) );
double* p_A = NUMERIC_POINTER(A);

我需要将此矩阵传递给C ++函数,其中类型vector<vector<double>>的变量'data'将使用矩阵A进行初始化。

我似乎无法弄清楚如何做到这一点。我想以更复杂的方式思考,我应该这样做,我打赌有一种简单的方法可以做到这一点。

2 个答案:

答案 0 :(得分:5)

您可能想要使用Rcpp。该软件包允许轻松集成R和C ++,包括将对象从R传递到C ++。该软件包可在CRAN上获得。此外,CRAN上的一些软件包使用Rcpp,因此它们可以作为灵感来源。 Rcpp的网站在这里:

http://dirk.eddelbuettel.com/code/rcpp.html

其中包括一些教程。

答案 1 :(得分:5)

正如保罗所说,我建议使用Rcpp来做这类事情。但这也取决于你想要vector< vector<double> >的意思。假设您要存储列,您可以像这样处理矩阵:

require(Rcpp)
require(inline)

fx <- cxxfunction( signature( x_ = "matrix" ), '
    NumericMatrix x(x_) ;
    int nr = x.nrow(), nc = x.ncol() ;
    std::vector< std::vector<double> > vec( nc ) ;
    for( int i=0; i<nc; i++){
        NumericMatrix::Column col = x(_,i) ;
        vec[i].assign( col.begin() , col.end() ) ;
    }
    // now do whatever with it
    // for show here is how Rcpp::wrap can wrap vector<vector<> >
    // back to R as a list of numeric vectors
    return wrap( vec ) ;
', plugin = "Rcpp" )
fx( A )
# [[1]]
# [1] 9 1
# 
# [[2]]
# [1] 3 6