Перевести код с С++ на Питон

#include <iostream>
#include <iomanip>
 
int main()
{
    int n;
    std::cin >> n;
    float** arr = new float* [n];
    float max;
    int i_max, j_max;
    for (int i = 0; i < n; ++i)
    {
        arr[i] = new float [n];
        for (int j = 0; j < n; ++j)
        {
            std::cin >> arr[i][j];
            if (((!i) && (!j)) || (arr[i][j] > max))
            {
                max = arr[i][j];
                i_max = i;
                j_max = j;
            }
        }
    }
    arr[i_max][j_max] = arr[0][0]; 
    arr[0][0] = max;
    for (int count = 1; count < n; ++count)
    {
        max = arr[0][1];
        for (int i = 0; i < n; ++i)
            for (int j = 0; j < n; ++j)
                if (((i != j) || ((i >= count) && (j >= count))) && (arr[i][j] > max))
                {
                    max = arr[i][j];
                    i_max = i;
                    j_max = j;
                }
        arr[i_max][j_max] = arr[count][count]; 
        arr[count][count] = max;
    }
    for (int i = 0; i < n; ++i)
    {
            for (int j = 0; j < n; ++j)
                std::cout << std::setw(5) << arr[i][j];
            std::cout << std::endl;
    }
    for (int i = 0; i < n; ++i)
    {
        int count = 0;
            for (int j = 0; j < n; ++j)
                if (arr[i][j] < 0)
                    ++count;
            if (count == n)
            {
                std::cout << "line #" << i + 1 << " has no positive numbers\n";
                system("pause");
                return 0;
            }
    }
    std::cout << "all lines have positive numbers\n";
    system("pause");
    return 0;
}