Numpy Construct Diagonal Matrix Definition