Question

在R中，我们可以使用is.complex（例如vec1=c(1+1i,2)）检查向量（例如is.complex(vec1)）是否包含复数。我想知道RcppArmadillo中的等效功能是什么？

以及如何像R中的Re(vec1)一样提取RcppArmadillo中向量中每个元素的实部？

Answer 1

要提取实部和虚部，可以使用arma::real()和arma::imag()函数。另外，您可以使用Sugar函数Rcpp::Re()和Rcpp::Im()：

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

// [[Rcpp::export]]
arma::vec getRe(arma::cx_vec x) {
  return arma::real(x);
}

// [[Rcpp::export]]
Rcpp::NumericVector getIm(Rcpp::ComplexVector x) {
  return Rcpp::Im(x);
}

/*** R
set.seed(42)
N <- 5
vec <- complex(5, rnorm(5), rnorm(5))

t(getRe(vec))
#>            [,1]        [,2]      [,3]       [,4]       [,5]
#> [1,] -0.9390771 -0.04167943 0.8294135 -0.4393582 -0.3140354
Re(vec)
#> [1] -0.93907708 -0.04167943  0.82941349 -0.43935820 -0.31403543

getIm(vec)
#> [1] -2.1290236  2.5069224 -1.1273128  0.1660827  0.5767232
Im(vec)
#> [1] -2.1290236  2.5069224 -1.1273128  0.1660827  0.5767232
*/

如果您使用上面的getRe(arma::vec x)，则会得到：

Warning message:
In getRe(vec) : imaginary parts discarded in coercion

您不能将复数放在一个不打算存储复数的对象中。这是C ++是强类型语言的结果。因此，无需使用is.complex()的类似物。

请参见Armadillo documentation，以获取更多参考。

在Rcpp（和RcppArmadillo）中，如何检查vec是否包含复数？

1 个答案: