Create N X N Matrix Python