Algorithm


Problem Name: Python - Eye and Identity

Problem Link: https://www.hackerrank.com/challenges/np-eye-and-identity/problem?isFullScreen=true  

In this HackerRank Functions in PYTHON problem solution,

identity

The identity tool returns an identity array. An identity array is a square matrix with all the main diagonal elements as 1 and the rest as

. The default type of elements is float.

 

import numpy
print numpy.identity(3) #3 is for  dimension 3 X 3

#Output
[[ 1.  0.  0.]
 [ 0.  1.  0.]
 [ 0.  0.  1.]]

 

eye

The eye tool returns a 2-D array with 1's as the diagonal and 0s elsewhere. The diagonal can be main, upper or lower depending on the optional parameter k.A . A positive k s for the upper diagonal, a negative k is for the lower, and a 0 k (default) is for the main diagonal.

import numpy
print numpy.eye(8, 7, k = 1)    # 8 X 7 Dimensional array with first upper diagonal 1.

#Output
[[ 0.  1.  0.  0.  0.  0.  0.]
 [ 0.  0.  1.  0.  0.  0.  0.]
 [ 0.  0.  0.  1.  0.  0.  0.]
 [ 0.  0.  0.  0.  1.  0.  0.]
 [ 0.  0.  0.  0.  0.  1.  0.]
 [ 0.  0.  0.  0.  0.  0.  1.]
 [ 0.  0.  0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.  0.]]

print numpy.eye(8, 7, k = -2)   # 8 X 7 Dimensional array with second lower diagonal 1.

 


 

Task

Your task is to print an array of size N * M with its main diagonal elements as 1's and 0's everywhere else.

Note

In order to get alignment correct, please insert the line numy.set_prinoptions(legacy = '1.13') below the numpy import.

Input Format

A single line containing the space separated values of N and M .

N denotes the rows.

M denotes the columns.

Output Format

Print the desired N * M

array.

Sample Input

3 3

Sample Output

[[ 1.  0.  0.]
 [ 0.  1.  0.]
 [ 0.  0.  1.]]

 

 

 

 

 

Code Examples

#1 Code Example with Python Programming

Code - Python Programming


import numpy as np

np.set_printoptions(legacy='1.13')

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

a = np.eye(n,m)

print(a)
Copy The Code & Try With Live Editor
Advertisements

Demonstration


Previous
[Solved] Zeros and Ones in PYTHON solution in Hackerrank
Next
[Solved] Merging Communities solution in Hackerrank - Hacerrank solution C, C++, java,js, Python