Question

我需要C ++等效于Python的scipy.sparse.rand函数（https://docs.scipy.org/doc/scipy/reference/generated/scipy.sparse.rand.html）。我将输入m, n, density参数，而rand()函数将为我生成一个矩阵（最好是COO矩阵）。我该怎么做？

Answer 1

如果您手上有boost，最好结合使用boost-ublas和boost-random来完成这项工作。为此，您必须预先安装boost库。

如果事先知道非零元素的数量，则可以优化代码。

您可以在以下位置观看代码的工作： https://coliru.stacked-crooked.com/a/097ea92bb336c774

#include <boost/numeric/ublas/matrix_sparse.hpp>
#include <boost/numeric/ublas/io.hpp>
#include <boost/random/mersenne_twister.hpp>
#include <boost/random/uniform_01.hpp>

int main() {
    int rows{ 4 };
    int cols{ 4 };
    double density{ 0.25 };
    boost::random::uniform_01<> dist;
    boost::random::mt19937 gen;

    boost::numeric::ublas::mapped_matrix<double> m(rows, cols, 3 * 3);
    for (unsigned i = 0; i < m.size1(); ++i)
        for (unsigned j = 0; j < m.size2(); ++j)
            if (dist(gen)>density) m(i, j) = dist(gen);

    std::cout << m << std::endl;
}

C ++等同于Python的scipy.sparse.rand

1 个答案: