Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Eigen: mask an array

Tags:

c++

eigen

Is it possible to mask an array in Eigen as in Matlab?

Something like

ArrayXd arrayA = ArrayXd::Random(10, 5);
ArrayXi mask = ArrayXi::Zero(arrayA.rows(), arrayA.cols());
mask = arrayA > 5;
ArrayXd arrayB = arrayA(mask)

where arrayB is a row vector containing all and only the elements of arrayA>5

I could find similar requests but not any updated answer after 2011 ( https://forum.kde.org/viewtopic.php?f=74&t=98382 , https://forum.kde.org/viewtopic.php?f=74&t=98093 , https://forum.kde.org/viewtopic.php?f=74&t=97652)

like image 922
lib Avatar asked Jul 09 '14 13:07

lib


2 Answers

From the Quick Reference

(R.array() < s).select(P,Q);  // (R < s ? P : Q)

so, in your case it would be

(arrayA > 5).select(mask, arrayA)
like image 181
user2658323 Avatar answered Sep 20 '22 16:09

user2658323


I found some reference here http://igl.ethz.ch/projects/libigl/matlab-to-eigen.html

For B = IM(A), they suggest:

B = A.unaryExpr(bind1st(mem_fun( 
    static_cast<VectorXi::Scalar&(VectorXi::*)(VectorXi::Index)>
    (&VectorXi::operator())), &IM)).eval();

But it's not a generalized solution (and actually I couldn't try it )

like image 31
lib Avatar answered Sep 23 '22 16:09

lib