benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 727 / How can I extract the main diagonal(1-d array) of a sparse matrix? The matrix is…

Problem

Answer published by the source. Consult the official source to check your work against its answer.

prompt

Problem:
How can I extract the main diagonal(1-d array) of a sparse matrix? The matrix is created in scipy.sparse. I want equivalent of np.diagonal(), but for sparse matrix.

A:
<code>
import numpy as np
from scipy.sparse import csr_matrix

arr = np.random.rand(4, 4)
M = csr_matrix(arr)
</code>
result = ... # put solution in this variable
BEGIN SOLUTION
<code>

library

Scipy

Discussion

Discussion

No discussion posts on this page yet. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

See answer Answer published by the source

Artifacts

Code, notes and reproducible work shared by participants. Files are served from a separate origin.

No artifacts on this page yet. Share reproducible code or notes in a contribution. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

Source and history

upstream_id / Original problem ID
727
task_id
9c5dacd7-8a49-53a8-8917-75143de6cede
task_key
default--test--727
task_revision_id
1
config / split
default / test

Official source

initial import