c++ - Eigen bool 数组切片

Question

在 MATLAB 中，从矩阵/数组(称为 logical indexing)中切出满足某些条件的值是很常见的。

vec = [1 2 3 4 5];
condition = vec > 3;
vec(condition) = 3;

我如何在 Eigen 中执行此操作？到目前为止，我有:

Eigen::Matrix<bool, 1, 5> condition = vec.array() > 3;

Answer 1

尝试 this :

#include <iostream>
#include <Eigen/Dense>

int main()
{
    Eigen::MatrixXi m(1, 5);
    m << 1, 2, 3, 4, 5;
    m = (m.array() > 3).select(3, m);
    std::cout << m << std::endl;

    return 0;
}