Diagonal Matrix Python Without Numpy