convert a numpy array to latex array

arraytotex.py

def totex(A):
    m, n = A.shape
    rstr = "".join(['r' for i in range(n)])
    print('\\left[\\begin{array}{' + rstr + r'}')
    for i in range(m):
        for j in range(n):
            if A[i,j] == 0:
                print(' ', end='')
            else:
                a = A[i,j]
                if float(a) == float(int(a)):
                    a = int(a)
                print(f'{a}', end='')
            if j != (n-1):
                print(' & ', end='')
            else:
                if i != (m-1):
                    print(' \\\\')
                else:
                    print('')
    print('\\end{array}\\right]')