2D向量的子向量

时间:2019-11-02 08:35:15

标签: c++ slice stdvector

我有std::vector<std::vector<double>> A(300, std::vector<double>(500))

我想创建一个新向量,其子范围为Asub-vector[5:10][25:100]。 我该怎么办?

4 个答案:

答案 0 :(得分:1)

您可以使用迭代器。首先创建向量:

std::vector<std::vector<double>> sub_vector;
sub_vector.reserve(5);

然后使用vector的范围构造函数填充它:

for (std::size_t i = 5; i < 10; ++i) {
    sub_vector.emplace_back(A[i].begin() + 25, A[i].begin() + 100);
}

注意:

  • 您有责任确保索引在范围内。否则会导致未定义的行为。

  • 这涉及开盘价范围。如果您要关闭范围,则需要在结束索引中添加一个。

答案 1 :(得分:1)

没有这样简单的符号,您必须自己动手:

FALSE

答案 2 :(得分:0)

另一个解决方案

使用函数来完成这项工作。

#include <iostream>
#include <vector>
#include <iterator>
#include <algorithm>

constexpr size_t MaxRows = 300;
constexpr size_t MaxColumns = 500;
using MyType = double;
using Columns = std::vector<MyType>;
using Matrix = std::vector<Columns>;

void copySubMatrix( const Matrix& source,Matrix& destination,const size_t& startRow,const size_t& endRow,const size_t& startColumn, const size_t& endColumn)
{
    // Clear destination matrix
    destination.clear();
    // Copy rows end columns
    std::for_each(source.begin() + startRow, source.begin() + endRow + 1, [&](const Columns & c) {
        Columns row{ c.begin() + startColumn, c.begin() + endColumn + 1};
        destination.push_back(row); });
}

int main() {

    // Define source matrix with given size and empty destination matrix
    Matrix A(MaxRows, Columns(MaxColumns));
    Matrix result{};

    // Fill source matrix with running values
    std::for_each(A.begin(), A.end(), [i = 0](Columns & c) mutable {for (MyType& m : c) m = i++; });


    // Copy the given range to the destination matrix
    copySubMatrix(A, result, 5, 10, 25, 100);

    // Display destination matrix
    std::for_each(result.begin(), result.end(), [](const Columns & c) {
        std::copy(c.begin(), c.end(), std::ostream_iterator<MyType>(std::cout, " ")); std::cout << "\n"; });

    return 0;
}

答案 3 :(得分:0)

您可以通过迭代一个维度来做到这一点:

std::vector<vector<double>> main_vector(300, std::vector<double>(500));
std::vector<vector<double>> sub_vector;
std::vector<double>::const_iterator first, last;

unsigned int x_pos_start=5, x_pos_end=10;
unsigned int y_pos_start=25, y_pos_end=100;

sub_vector.resize(x_pos_end - x_pos_start + 1);
for(size_t i=x_pos_start; i<=x_pos_end; ++i)
{
    first = main_vector[i].begin() + y_pos_start;
    last = main_vector[i].begin() + y_pos_end;
    sub_vector[i].insert(sub_vector[i].begin(), first, last);
}