Algorithm


Problem Name: Python - Transpose and Flatten

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

In this HackerRank Functions in PYTHON problem solution,

Transpose

 

We can generate the transposition of an array using the tool numpy.transpose.
It will not affect the original array, but it will create a new array.

 

import numpy

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

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

 

Flatten

 

The tool flatten creates a copy of the input array flattened to one dimension.

 

import numpy

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

#Output
[1 2 3 4 5 6]

 


 

Task

You are given a N * M integer array matrix with space separated elements (N = rows and M = columns).
Your task is to print the transpose and flatten results.

Input Format

The first line contains the space separated values of N and M .
The next N lines contains the space separated elements of M columns.

Output Format

First, print the transpose array and then print the flatten.

Sample Input

2 2
1 2
3 4

Sample Output

[[1 3]
 [2 4]]
[1 2 3 4]

 

 

 

 

Code Examples

#1 Code Example with Python Programming

Code - Python Programming


import numpy as np

# Take input in the form of list, here each value in input().split() is being converted into int() using list comprehensible

n,m = [int(x) for x in input().split()]

arr = []

# map function runs the iterable values of the second argument through the first function which is int() here.

for i in range(n):
    row = list(map(int, input().split()))
    arr.append(row)

arr = np.array(arr)
print(np.transpose(arr))
print(arr.flatten())
Copy The Code & Try With Live Editor
Advertisements

Demonstration


Previous
[Solved] Shape and Reshape in PYTHON solution in Hackerrank
Next
[Solved] Concatenate in PYTHON solution in Hackerrank