c++方法按列中的元素对矩阵进行排序(与MATLAB中的sortrows相同)

Question

我正在寻找一种 C++ 方法，它能够按照特定列中值的降序对矩阵的行进行排序。

例如，如果这是我的输入:

matrix = [1,2,3;
          2,4,1;
          0,5,2]

方法(matrix,3) 的调用应提供以下输出:

outputMatrix = [1,2,3;
                0,5,2;
                2,4,1]

在 MATLAB 中，这可以通过调用函数来完成:

outputMatrix = sortrows(matrix,3).

C++ 呢？ (在这两种情况下，3 都是列的索引)。

此外，在我正在使用的脚本中，矩阵被定义为 vector 的 vector :std::vector<std::vector<double> > matrix

[编辑] 我添加另一个示例:

input = [4,5,6;
         0,2,8;
         1,2,3;
         6,7,9]

输出矩阵 = 排序行(输入，2)；第 2 列，有序，是:9,8,6,3；所以我必须对行进行排序并复制前两列的元素(分别为 0 和 1)。

 outputMatrix = [6,7,9;
                 0,2,8;
                 4,5,6;
                 1,2,3]

我在这里报告我写的方法，但我不知道这是否是一种快速的方法:

std::vector<std::vector<double> > sortrows(std::vector<std::vector<double> > matrix,int col){
    int length = matrix[col].size();
    std::vector<std::vector<double> > output(3,std::vector<double>(length*length));
    output[col] = matrix[col];
    std::sort(output[col].begin(),output[col].end(),std::greater<double>());
    for (int i = 0; i < length*length;i++){
        int index = 0;
        while(output[col][i]!=matrix[col][index]){index++;}
        output[0][i]=matrix[0][index];
        output[1][i]=matrix[1][index];
        matrix[2][index] = -1;
    }
    return output;
}

Answer 1

类似于下面(在 C++11 中)，根据特定列对矩阵行进行排序

void sortrows(std::vector<std::vector<double>>& matrix, int col) {    
    std::sort(matrix.begin(),
              matrix.end(),
              [col](const std::vector<double>& lhs, const std::vector<double>& rhs) {
                  return lhs[col] > rhs[col];
              });
}

另请注意，此代码更改原始矩阵而不是返回新矩阵。

非 C++11 版本:

class Compare {
public:
    Compare(int col) : col_(col) {}
    bool operator()(std::vector<double>& lhs, std::vector<double>& rhs) {
        return lhs[col_] > rhs[col_];
    }
private:
    int col_; 
};

void sortrows(std::vector<std::vector<double>>& matrix, int col) {    
    std::sort(matrix.begin(), matrix.end(), Compare(col));
}