Algorithm


Problem Name: Python - Shape and Reshape

Problem Link: https://www.hackerrank.com/challenges/np-shape-reshape/problem?isFullScreen=true  

In this HackerRank Functions in PYTHON problem solution,

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 * 3 NumPy array.

Input Format

A single line of input containing 9 space separated integers.

Output Format

Print the 3 * 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]]

 

 

 

 

Code Examples

#1 Code Example with Python Programming

Code - Python Programming


import numpy

i = input().strip().split(' ')

i = numpy.array(i).reshape(3,3).astype(int)

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

Demonstration


Previous
[Solved] Arrays in PYTHON solution in Hackerrank
Next
[Solved] Transpose and Flatten in PYTHON solution in Hackerrank