benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 477 / I am using Python with numpy to do linear algebra.

Problem

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

prompt

Problem:
I am using Python with numpy to do linear algebra.
I performed numpy SVD on a matrix `a` to get the matrices U,i, and V. However the i matrix is expressed as a 1x4 matrix with 1 row. i.e.: [ 12.22151125 4.92815942 2.06380839 0.29766152].
How can I get numpy to express the i matrix as a diagonal matrix like so: [[12.22151125, 0, 0, 0],[0,4.92815942, 0, 0],[0,0,2.06380839,0 ],[0,0,0,0.29766152]]
Code I am using:
a = np.matrix([[3, 4, 3, 1],[1,3,2,6],[2,4,1,5],[3,3,5,2]])
U, i, V = np.linalg.svd(a,full_matrices=True)
So I want i to be a full diagonal matrix. How an I do this?
A:
<code>
import numpy as np
a = np.matrix([[3, 4, 3, 1],[1,3,2,6],[2,4,1,5],[3,3,5,2]])
U, i, V = np.linalg.svd(a,full_matrices=True)
</code>
i = ... # 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
477
task_id
82dec25f-929e-51c2-8032-d3d127d636fc
task_key
default--test--477
task_revision_id
1
config / split
default / test

Official source

initial import