HackerRank Python Solution - Numpy Topic - Shape and Reshape

Shape:

The shape tool gives a tuple of array dimensions and can be used to change the dimensions of an array. 

(a). Using shape to get array dimensions

import numpy

my__1D_array = numpy.array([1, 2, 3, 4, 5])
print my_1D_array.shape     #(5,) -> 1 row and 5 columns

my__2D_array = numpy.array([[1, 2],[3, 4],[6,5]])
print my_2D_array.shape     #(3, 2) -> 3 rows and 2 columns 

(b). Using shape to change array dimensions

import numpy

change_array = numpy.array([1,2,3,4,5,6])
change_array.shape = (3, 2)
print change_array      

#Output
[[1 2]
[3 4]
[5 6]]

Reshape:

The reshape tool gives a new shape to an array without changing its data. It creates a new array and does not modify the original array itself.

import numpy

my_array = numpy.array([1,2,3,4,5,6])
print numpy.reshape(my_array,(3,2))

#Output
[[1 2]
[3 4]
[5 6]]

Task:

You are given a space-separated list of nine integers. Your task is to convert this list into a 3 x 3 NumPy array. 

Input Format:

A single line of input containing space-separated integers.

Output Format:

Print the 3 x 3 NumPy array.

Sample Input:

1 2 3 4 5 6 7 8 9

Sample Output:

[[1 2 3]
 [4 5 6]
 [7 8 9]]

Solution:

import numpy

arr = numpy.array(list(map(int,input().split())))

reshaped_arr = numpy.reshape(arr,(3,3))

print(reshaped_arr)

Disclaimer: The problem statement is given by hackerrank.com but the solution is generated by the Geek4Tutorial's author. If there is any concern regarding this post or website, please contact us using the contact form. Thank you!