benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 478 / What is the quickest way to convert the non-diagonal elements of a square…

Problem

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

prompt

Problem:
What is the quickest way to convert the non-diagonal elements of a square symmetrical numpy ndarray to 0? I don't wanna use LOOPS!
A:
<code>
import numpy as np
a = np.array([[1,0,2,3],[0,5,3,4],[2,3,2,10],[3,4, 10, 7]])
</code>
a = ... # put solution in this variable
BEGIN SOLUTION
<code>

library

Numpy

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
478
task_id
df2d977b-9b49-55bf-995b-6f2e89f6340d
task_key
default--test--478
task_revision_id
1
config / split
default / test

Official source

initial import