c++ - Eigen :掩码数组

Question

是否可以像在 Matlab 中那样在 Eigen 中屏蔽数组？

有点像

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

其中 arrayB 是包含 arrayA 的所有且仅包含元素的行 vector >5

我可以找到类似的请求，但在 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)

Answer 1

来自Quick Reference

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

所以，在你的情况下会是

(arrayA > 5).select(mask, arrayA)