Python solutions 040 LK

2 files changed, 49 insertions, 0 deletions

diff --git a/challenge-040/lubos-kolouch/python/ch-1.py b/challenge-040/lubos-kolouch/python/ch-1.py
new file mode 100644
index 0000000000..bebdf0e5dc
--- /dev/null
+++ b/challenge-040/lubos-kolouch/python/ch-1.py
@@ -0,0 +1,20 @@
+#!python3
+
+import numpy as np
+import sys
+
+# python ch-1.py "[ I L O V E Y O U ][ 2 4 0 3 2 0 1 9 ][ ! ? £ $ % ^ & * ]"
+
+assert len(sys.argv) == 2
+
+arr = np.empty(0)
+arr1 = np.array(sys.argv[1].split(' ]'))
+
+for elem in arr1[:-1]:
+    arr = np.append(arr, elem[2:].split(' '))
+
+arr = np.reshape(arr, (len(arr1)-1, int(len(arr)/(len(arr1)-1))))
+arr = np.rot90(arr, 3)
+arr = np.flip(arr, 1)
+
+print(arr)
diff --git a/challenge-040/lubos-kolouch/python/ch-2.py b/challenge-040/lubos-kolouch/python/ch-2.py
new file mode 100644
index 0000000000..3c0a6706bb
--- /dev/null
+++ b/challenge-040/lubos-kolouch/python/ch-2.py
@@ -0,0 +1,29 @@
+#!python3
+
+import numpy as np
+import sys
+
+# python ch-2.py "[ 10, 4, 1, 8, 12, 3 ][ 0, 2, 5 ]"
+
+assert len(sys.argv) == 2
+
+arr1 = np.empty(0, int)
+arr2 = np.empty(0, int)
+arr = np.array(sys.argv[1].replace('[', '').replace(' ', '').split(']'))
+
+arr1 = arr[0].split(',')
+arr2 = arr[1].split(',')
+
+arr_sort = np.empty(0, int)
+
+for key in arr2:
+    arr_sort = np.append(arr_sort, int(arr1[int(key)]))
+
+arr_sort = np.sort(arr_sort)
+
+i = -1
+for key in arr2:
+    i += 1
+    arr1[int(key)] = arr_sort[i]
+
+print(arr1)