How to create diagonal array from existing array in numpy

Viewed 341

I am trying to make a diagonal numpy array from:

[1,2,3,4,5,6,7,8,9]

Expected result:

[[ 0,  0,  1,  0,  0],
 [ 0,  0,  0,  2,  0],
 [ 0,  0,  0,  0,  3],
 [ 4,  0,  0,  0,  0],
 [ 0,  5,  0,  0,  0],
 [ 0,  0,  6,  0,  0],
 [ 0,  0,  0,  7,  0],
 [ 0,  0,  0,  0,  8],
 [ 9,  0,  0,  0,  0]]

What would be an efficient way of doing this?

2 Answers
Related