PYTHONДано простое число. Составить функцию, которая будет находить следующее за ним простое число.

Код на C++

#include <iostream>

using namespace std;

int main()

{

int n;

cin >> n;

for (int i = 0; i < n; i++) {

for (int j = 0; j < n; j++) {

if ((i % 2 == 0 && j % 2 == 0)||(i % 2 != 0 && j % 2 !=0)) {

cout << '. ';

} else {

cout << "°";

}

}

cout << endl;

}

}