Numpy!

snehalmastud · Feb 23, 2020 · 8ca5a03 · 8ca5a03
1 parent f80d4f2
commit 8ca5a03
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Numpy/Transpose and Flatten/Solution.py b/Numpy/Transpose and Flatten/Solution.py
@@ -0,0 +1,10 @@
+import numpy
+
+n, m = raw_input().split()
+lines = []
+for _ in range(int(n)):
+    lines.append(map(int, raw_input().split()))
+
+my_array = numpy.array(lines)
+print numpy.transpose(my_array)
+print my_array.flatten()