4

我一生都无法弄清楚为什么这不能正常工作。它似乎没有返回第 k 个元素。

typedef Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic> matrix;

double test(matrix& D, int k)
{
    auto d = D.row(1);
    std::nth_element(d.data(),d.data()+k, d.data()+d.size());
    return d(k) ;
}

我也试过

template <typename ScalarType, typename Derived>
void Sort(Eigen::MatrixBase<Derived> &xValues)
{
std::sort(xValues.derived().data(), xValues.derived().data()+xValues.derived().size());
}


double test(matrix& D, int k)
{
    auto d = D.row(1);
    Sort<double>(d);
    return d(k) ;
}

非常感谢任何帮助。

编辑：-

我刚刚尝试改变

auto d = D.row(1);

到

Eigen::VectorXd rowD = D.row(1);
....

它似乎工作正常。

对此有些困惑。

4

1 回答 1

6

默认情况下，特征矩阵是列优先的。这意味着，矩阵的一行不是连续的 C 数组，您不能将数据指针用作迭代器。

例如，3x4 矩阵将存储为：

0 3 6 9
1 4 7 10
2 5 8 11

现在，row(1)将是

1 4 7 10

但是您传递给的指针迭代器nth_element()将访问

1 2 3 4

如果您将matrixtypedef 更改为 row-major，则您的代码有效：

typedef Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> matrix;

更新：您编辑的示例有效，因为您将行复制到向量。对于向量（一维矩阵），数据是按行优先还是按列存储并不重要。

于 2012-04-13T12:29:02.900 回答