如何计算CSR矩阵对角块中每行的非零数 我有一个CSR格式的矩阵，需要一个C++向量，它包含每行的非零条目（计数），限制为正方形，不同大小的对角线块。 // The matrix in CSR format std::vector<int> row_idx = {0,2,4,6,10,13}; // size n+1 where 0-n are the idx of the row starts in values and column_idx and n+1 the TOTAL number of values std::vector<int> values = {1,6,2,7,3,8,10,11,4,9,12,13,5}; // nonzero matrix values std::vector<int> column_idx = {0,3,1,3,2,4,0,1,3,4,2,3,4}; // column indices of the values //CSR格式的矩阵 std:：vector row_idx={0,2,4,6,10,13}；//大小n+1，其中0-n是行的idx以值开始，列为_idx，n+1是值的总数 向量值={1,6,2,7,3,8,10,11,4,9,12,13,5}；//非零矩阵值 std：：向量列_idx={0,3,1,3,2,4,0,1,3,4,2,3,4}；//值的列索引

下面的示例有两个大小不同的块A和B（感兴趣的块始终是正方形和对角线）

本示例所需的结果将是NnZGuiin范围[n]＝{1,1,2，2，3}，但由于它需要嵌入另一个例程中，我主要在寻找一个例程，用C++计算各个块。大概是这样的：

// block A
int rangeStart = 0;
int rangeEnd = 2;

// block B
//int rangeStart = 2;
//int rangeEnd = n;

for (int i = rangeStart; i<rangeEND; ++i)
{
    nnz_in_ranges[n] = ...
}

// desired result for block A: nnz_in_ranges[n] = {1,1,0,0,0}
// desired result for block B: nnz_in_ranges[n] = {0,0,2,2,3}

std::vector<int> count_positions_in_block(int block_begin, int block_size, 
    const std::vector<int>& row_idx, const std::vector<int>& column_idx)
{
    std::vector<int> cnt(block_size, 0);

    const auto block_end = block_begin + block_size;
    assert(block_end < row_idx.size());

    for (auto row = block_begin; row < block_end; ++row)
    {
        auto first = row_idx[row];
        auto last = row_idx[row + 1];
        assert(first <= last);

        for (auto i = first; i < last; ++i)
            if (column_idx[i] >= block_begin && column_idx[i] < block_end)
                ++cnt[row - block_begin];
    }

    return cnt;
}

auto nnz1 = count_positions_in_block(0, 2, row_idx, column_idx);
// nnz1 = [1, 1]
auto nnz2 = count_positions_in_block(2, 3, row_idx, column_idx);
// nnz2 = [2, 2, 3]

//块A
int rangeStart=0；
int rangeEnd=2；
//B区
//int rangeStart=2；
//int rangeEnd=n；
对于（int i=rangeStart；i稀疏矩阵可以显式包含零个条目。从您的问题来看，不清楚您是要计算位置还是实际值。我将假设前者，因为您的计算代码不使用
值

然后我们按照CSR格式的定义得到如下内容：

// block A
int rangeStart = 0;
int rangeEnd = 2;

// block B
//int rangeStart = 2;
//int rangeEnd = n;

for (int i = rangeStart; i<rangeEND; ++i)
{
    nnz_in_ranges[n] = ...
}

// desired result for block A: nnz_in_ranges[n] = {1,1,0,0,0}
// desired result for block B: nnz_in_ranges[n] = {0,0,2,2,3}

std::vector<int> count_positions_in_block(int block_begin, int block_size, 
    const std::vector<int>& row_idx, const std::vector<int>& column_idx)
{
    std::vector<int> cnt(block_size, 0);

    const auto block_end = block_begin + block_size;
    assert(block_end < row_idx.size());

    for (auto row = block_begin; row < block_end; ++row)
    {
        auto first = row_idx[row];
        auto last = row_idx[row + 1];
        assert(first <= last);

        for (auto i = first; i < last; ++i)
            if (column_idx[i] >= block_begin && column_idx[i] < block_end)
                ++cnt[row - block_begin];
    }

    return cnt;
}

auto nnz1 = count_positions_in_block(0, 2, row_idx, column_idx);
// nnz1 = [1, 1]
auto nnz2 = count_positions_in_block(2, 3, row_idx, column_idx);
// nnz2 = [2, 2, 3]

值中的所有矩阵元素是否总是非零？有时一些零被显式存储。是否只计算“位置”不管这些位置中的实际值是多少？我想计算位置。实际上我想计算位置。如果在lambda函数中需要常量，我尝试了计数，但无法为计数范围引入变量。@klaubs，添加了
count\u if
ied版本。