Thank you for the solution, below easy method also works fine:

n, m = input().split()

numpy.set_printoptions(sign=' ') print(numpy.eye(int(n), int(m)))