Rcpparmadillo C++ 创建布尔向量
Rcpparmadillo c++ create bool vector
我正在尝试使用 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
将逻辑向量作为参数传递的最佳方法是什么?
您想要 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>
我正在尝试使用 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
将逻辑向量作为参数传递的最佳方法是什么?
您想要 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>