Diagonal Matrix Python Without Coding