RcppEigen查找某元素是否在eigen::vector，eigen::matrix

最新推荐文章于 2023-11-12 09:43:27 发布

zaza0_0

最新推荐文章于 2023-11-12 09:43:27 发布

阅读量992

点赞数 1

分类专栏： R语言

本文链接：https://blog.csdn.net/zaza0_0/article/details/104581400

版权

R语言专栏收录该内容

22 篇文章 0 订阅

订阅专栏

//[[Rcpp::depends(RcppEigen)]]
#include<RcppEigen.h>
#include<Rcpp.h>
using namespace std;
using namespace Eigen;

// [[Rcpp::export]]
double testcwiseEqual(VectorXd x,double y){
  //说明：A.cwiseEqual(y),A,y可以是向量，矩阵，数值。若y为向量，不管是否只有一个元素，只与A的元素依次比较，一个元素时只比第一个元素。若y为数值，则与A所有元素比较
  
  VectorXd c;
  double s;
  c=x.cwiseEqual(y).cast<double>();//本来返回的是布尔值矩阵，要转为double
  s=c.sum();
  cout<<"sum"<<x.cwiseEqual(y).sum()<<endl;
  cout<<x.cwiseEqual(y)<<endl;
return s;
  
  }

在这里插入图片描述

MatrixXi m(2,2);
m << 1, 0,
     1, 1;
cout << "Comparing m with identity matrix:" << endl;
cout << m.cwiseEqual(MatrixXi::Identity(2,2)) << endl;
Index count = m.cwiseEqual(MatrixXi::Identity(2,2)).count();
cout << "Number of coefficients that are equal: " << count << endl;