Assign shifted diagonal values with np.diag

Question

I\'m trying to assign values to multiple diagonals of a matrix. For example, I have this matrix:

>>> u = np.ze