HackerRank Transpose and Flatten problem solution in python

In this Transpose and Flatten problem, You are given a N X M integer array matrix with space-separated elements (N = rows and M = columns). Your task is to print the transpose and flatten results.

Problem solution in Python 2 programming.

import numpy
N, M = map(int, raw_input().split())
s = numpy.array([map(int, raw_input().split()) for _ in range(N)])
print numpy.transpose(s)
print s.flatten()

Problem solution in Python 3 programming.

import numpy
n, m = map(int, input().split())
array = numpy.array([input().strip().split() for _ in range(n)], int)
print (array.transpose())
print (array.flatten())

Problem solution in pypy programming.

# Enter your code here. Read input from STDIN. Print output to STDOUT
import numpy

n,m = map(int, raw_input().split())
a = numpy.array([raw_input().split() for _ in xrange(n)], int)

print (a.transpose())
print (a.flatten())

Problem solution in pypy3 programming.

import numpy
narr = numpy.array([input().split() for _ in range(int(input().split()[0]))],int)
print (narr.transpose())
print (narr.flatten())