Как перебрать ненулевые значения разреженного массива

Question

У меня есть разреженный массив в Julia, и я хотел бы перебрать его ненулевые записи.Какой хороший способ сделать это?

Пока у меня есть:

using SparseArrays

a = sprandn(20,20,0.3)

for (x,y,v) in a
  print(x,y,v)
end

Answer 1

Вот прямой способ сделать это

using SparseArrays

function print_nz(A)
    for col in 1:size(A, 2)
        for r in nzrange(A, col)
            println(rowvals(A)[r], ' ', col, ' ',  nonzeros(A)[r])
        end
    end
end

и пример его использования:

julia> A = [1 2 1
            0 0 1
            3 0 0];

julia> print_nz(sparse(A))
1 1 1
3 1 3
1 2 2
1 3 1
2 3 1

Answer 2

Функция findnz возвращает кортеж, содержащий массивы компонентов x, y и value разреженной матрицы.То есть

findnz(a) gives ([x1, x2, x3, ...], [y1, y2, y3, ...], [v1, v2, v3, ...])

Вы можете использовать это как итератор следующим образом:

for (x,y,v) in zip(findnz(a)...)
  println(x,' ',y,' ',v)
end