import numpy as np ... shape = (6, 6) #This will be some pre-determined size sigma = np.diag(S) #diagonalise the matrix - this works sigma.resize(shape) #Resize the matrix and fill with zeros