使用迭代器遍历boost :: ublas矩阵

时间:2014-09-25 17:27:34

标签: c++ boost matrix iterator ublas

我只是想从头到尾遍历矩阵,触及每个元素。但是,我看到没有一个迭代器用于boost矩阵,而是有两个迭代器,我无法弄清楚如何使它们工作以便你可以遍历整个矩阵

    typedef boost::numeric::ublas::matrix<float> matrix;

    matrix m1(3, 7);

    for (auto i = 0; i < m1.size1(); i++)
    {
        for (auto j = 0; j < m1.size2(); j++)
        {
            m1(i, j) = i + 1 + 0.1*j;
        }
    }

    for (auto itr1 = m1.begin1(); itr1!= m1.end1(); ++itr1)
    { 
        for (auto itr2 = m1.begin2(); itr2 != m1.end2(); itr2++)
        {
            //std::cout << *itr2  << " ";
            //std::cout << *itr1  << " ";
        }
    }

我的这段代码,使用itr1只打印矩阵的第1行,使用itr2只打印矩阵的第1列。可以做些什么来访问所有行和列?

1 个答案:

答案 0 :(得分:8)

要迭代矩阵,应该从iterator1中获取iterator2,如下所示:

for(auto itr2 = itr1.begin(); itr2 != itr1.end(); itr2++)

完整代码:

#include <stdlib.h>
#include <boost/numeric/ublas/matrix.hpp>

using namespace boost::numeric::ublas;
using namespace std;

int main(int argc, char* argv[]) {

  typedef boost::numeric::ublas::matrix<float> matrix;

  matrix m1(3, 7);
  for (auto i = 0; i < m1.size1(); i++) {
    for (auto j = 0; j < m1.size2(); j++) {
      m1(i, j) = i + 1 + 0.1*j;
    }
  }

  for(matrix::iterator1 it1 = m1.begin1(); it1 != m1.end1(); ++it1) {
    for(matrix::iterator2 it2 = it1.begin(); it2 !=it1.end(); ++it2) {
      std::cout << "(" << it2.index1() << "," << it2.index2() << ") = " << *it2 << endl;
    }
    cout << endl;
  }

  return EXIT_SUCCESS;
}

输出:

(0,0) = 1
(0,1) = 1.1
(0,2) = 1.2
(0,3) = 1.3
...