import numpy as np

np.set_printoptions(legacy='1.13') n = input() l1 = list(map(int,n.split(' '))) print(np.eye(l1[0],l1[1]))

For Python3 Platform

import numpy

numpy.set_printoptions(legacy='1.13')

N, M = map(int, input().split())

print(numpy.eye(N, M))

For Python3

import numpy

numpy.set_printoptions(legacy="1.13")

n, m = map(int, input().split())

print(numpy.eye(n, m))

import numpy numpy.set_printoptions(legacy='1.13')

print(numpy.eye(*map(int,input().split())))

import numpy numpy.set_printoptions(legacy='1.13') N, M = map(int, input().strip().split()) print(numpy.eye(N, M))