numpy set diagonal

polski