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;

  }

}