Rcpparmadillo c++创建布尔向量。

Question

Rcpparmadillo c++创建布尔向量。

4

我想使用Rcpparmadillo将一个bool向量作为参数传递给一个函数。一个简单的例子如下：

// [[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;
}

这里建议使用mat<unsigned char>类型，但对我无效。我也尝试了uvec和std::vector<bool>，但它们都不起作用。在使用Rcpparmadillo时，传递一个逻辑向量作为参数的最佳方法是什么？此处有更多信息。

- nopeva

尝试以更标准的方式格式化您的代码。 - Dirk Eddelbuettel

也许这可以帮助：https://dev59.com/f3fZa4cB1Zd3GeqPMgnS - user4651282

1

row成员函数从矩阵返回一行向量。在这种情况下，您有一个向量，因此您希望mybool(i)返回一个带有边界检查的元素，或者使用mybool[i]而不进行边界检查。 - Manos Nikolaidis

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Dirk Eddelbuettel · Accepted Answer

您需要从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>