search

jmpark0118 / CODING_TEST_PRACTICE

0 stars 0 forks source link

Practice>Python>Numpy>Eye and Identity #109

Open jmpark0118 opened 3 years ago

jmpark0118 commented 3 years ago

image image

출처 : https://www.hackerrank.com/challenges/np-eye-and-identity/problem

jmpark0118 commented 3 years ago 

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

if __name__ == '__main__':
    n, m = map(int, input().split())
    if n==m:
        print(np.identity(n))
    else:
        print(np.eye(n, m))