构建一个新矩阵，由给定行索引向量的一些源矩阵行组成

Question

我想构建一个新矩阵，由一些源矩阵行组成，给定一个非连续索引的向量。

也就是说，我想要一个 row(( 函数，它接受索引列表并返回存储在新矩阵中的行列表：

  VectorXi v = VectorXi::LinSpaced( 4, 10, 13);
  MatrixXi m = v.rowwise().replicate( 4 );
  VectorXi r1 ( ( VectorXi(3) <<  0,  3, 1 ).finished() );
  // Here is some pseudo code to create the desired matrix N :
  MatrixXi N = m.row(r1);
  cout << "m = " << m << endl;
  cout << "r1 = " << r1 << endl;
  cout << "N = " << N << endl;

期望输出 ：

m = 
10 10 10 10
11 11 11 11
12 12 12 12
13 13 13 13
r1 =
0
3
1
N =
10 10 10 10
13 13 13 13
11 11 11 11

非常感谢您的帮助。

西尔万

Answer 1

启用开发分支和(至少(C++11 后，您可以编写：

Eigen::MatrixXi N = m(r1,Eigen::all);

这类似于 Matlab 语法：

N = m(r1, :);

您也可以直接传递{x,...} -lists，或任何行为类似于std::vector<int>的东西(必须提供size()函数和operator[]并返回整型(，例如：

std::vector<int> c2{{3,0}};
std::cout << "m({2,1},c2) = n" << m({2,1}, c2) << 'n';

这些表达式是可写的(假设m本身是可写的(：

m({2,1}, c2) = Eigen::Matrix2i{{1,2},{3,4}};
std::cout << m << 'n';

戈博尔特演示：https://godbolt.org/z/cjacOY