Practice>Python>Numpy>Transpose and Flatten

[jmpark0118]

출처 : https://www.hackerrank.com/challenges/np-transpose-and-flatten/problem

[jmpark0118]

import numpy

if __name__ == '__main__':
    n, m = map(int, input().split())

    my_list = []
    for i in range(n):
        my_list.append(list(map(int, input().split())))

    my_arr = numpy.array(my_list).reshape(n, m)

    print(my_arr.transpose())
    print(my_arr.flatten())