特征索引，更新特定行的所有列

Question

假设我有一个 ArrayXXf（或 MatrixXf）m。在 for 循环的每次迭代中，我想用 VectorXf.

按行填充 m

Eigen::ArrayXXf m(5, 5);

for (int i = 0; i < 5; i++)
{
    Eigen::VectorXf vec(5);
    vec << i, i + 1, i + 2, i+3, i+4;

    //fill m row wise
    // in matlab I will do something like m(i,:) = vec; 
    // in numpy this will looks like m[i:] = vec;
    // that means when i is 0 m looks like 
    //          [ 0 1 2 3 4 5
    //           -  - - - - -
    //           -  - - - - -
    //           -  - - - - -
    //           -  - - - - -]
}

如何在 Eigen 中实现？

Answer 1

使用 block() 函数。

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

using namespace std;

int main()
{
    Eigen::ArrayXXf m(5, 5);

    for (int i = 0; i < 5; i++) {
        Eigen::VectorXf vec(5);
        vec << i, i + 1, i + 2, i+3, i+4;

        m.block(i, 0, 1, 5) << vec.transpose();
    }

    std::cout << m << std::endl;
    return 0;
}

输出：

编辑：

还有一种更简单的替代方法：row() 函数。

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

using namespace std;

int main()
{
    Eigen::ArrayXXf m(5, 5);

    for (int i = 0; i < 5; i++) {
        Eigen::VectorXf vec(5);
        vec << i, i + 1, i + 2, i+3, i+4;

        m.row(i) = vec.transpose();
    }

    std::cout << m << std::endl;
    return 0;
}

输出：

P.S。 transpose() 是必需的，因为 Eigen::VectorXf 默认情况下是列向量，不是行向量。

Answer 2

为了简化@Kunal 的回答，您可以直接修改数组（或矩阵）的行（或列），而无需创建临时向量。在您的示例中，您可以使用 .setLinSpaced():

Eigen::ArrayXXf m(5, 5);

for (int i = 0; i < 5; i++) {
    m.row(i).setLinSpaced(i,i+4); //.col(i) would be slightly more efficient
}

或使用逗号初始值设定项：

for (int i = 0; i < 5; i++) {
    m.row(i) << i, i+1, i+2, i+3, i+4;
}

特征索引，更新特定行的所有列

Eigen indexing, update all column of a specific row

c++

eigen