np-eye-and-identity.py

# Python > Numpy > Eye and Identity
# Create an array using the identity or eye tools from the NumPy module.
#
# https://www.hackerrank.com/challenges/np-eye-and-identity/problem
#

import numpy

if numpy.version.version >= '1.14.':
    numpy.set_printoptions(legacy='1.13')

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

print(numpy.eye(n, m))