Rcpparmadillo c++ 创建布尔向量答案

【问题标题】：Rcpparmadillo c++ create bool vectorRcpparmadillo c++ 创建布尔向量
【发布时间】：2015-08-31 11:12:48
【问题描述】：

我正在尝试将 bool 向量作为参数传递给使用 Rcpparmadillo 的函数。一个愚蠢的例子是这样的：

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadillo.h>

// [[Rcpp::export]]
arma::mat myfun(arma::mat A, arma::vec mybool)
{
    int n = A.n_rows;
    arma::vec B(n);

    for(unsigned int i = 0; i < n; ++i)
    {
        if(mybool.row(i) && i < 10) // mybool.row(i) && throws the error
        {
            B.row(i) = arma::accu(A.row(i));
        }
        else
        {
            B.row(i) = pow(arma::accu(A.row(i)), 0.5);
        }
    }

    return B;
}

Here 建议使用 mat<unsigned char> 类型，但不适用于我。我也试过uvec 和std::vector<bool>，但也没有用。使用Rcpparmadillo 将逻辑向量作为参数传递的最佳方法是什么？

【问题讨论】：

尝试以更标准的方式格式化您的代码。
也许这有帮助：stackoverflow.com/questions/18899665/…
行成员函数从矩阵返回行向量。在这种情况下，您有一个向量，因此您希望 mybool(i) 返回一个带边界检查的元素或 mybool[i] 不带边界检查的元素。

标签： c++ r vector rcpp armadillo

【解决方案1】：

您想要来自 Armadillo 的 uvec - 它没有 bool 类型。这是您的代码的重新格式化版本 * 使用uvec * 直接索引向量

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadillo.h>

// [[Rcpp::export]]
arma::mat myfun(arma::mat A, arma::uvec mybool) {
    unsigned int n = A.n_rows;
    arma::vec B(n);
    for (unsigned int i=0; i<n; ++i) {
        if (mybool[i] && i < 10) {
           B[i] = arma::accu(A.row(i)) ;
        } else {
           B[i] = pow(arma::accu(A.row(i)), 0.5);
        }
    } //end loop
    return B;
}

/*** R
A <- matrix(1:16,4,4)
mybool <- c(FALSE, TRUE, TRUE, FALSE)
myfun(A, mybool)
*/

如果我们sourceCpp()这个，它会为我们运行底部的R：

R> sourceCpp("/tmp/ap13.cpp")

R> A <- matrix(1:16,4,4)

R> mybool <- c(FALSE, TRUE, TRUE, FALSE)

R> myfun(A, mybool)
         [,1]
[1,]  5.29150
[2,] 32.00000
[3,] 36.00000
[4,]  6.32456
R>

【讨论】：

需要注意的几点。 (1)。通过避免在开始时复制A 和mybook，可以提高该功能的效率。应该是arma::mat myfun(const arma::mat& A, const arma::uvec& mybool)。 (2) A 中的数据应该按列而不是按行存储和访问，因为犰狳以逐列的方式存储数据。
@mtall：你有 C/C++ 程序员的细心，但我邀请你介绍一下。我们的对象实际上是底层 SEXP 指针对象周围的瘦代理。 所以没有副本。您现在也可以通过const & 写信但性能将（大约）相同。