Diagonal Matrix Python Without Numpy Documentation