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)
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)
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 )
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With