python code for challenge 70

1 files changed, 14 insertions, 0 deletions

diff --git a/challenge-070/walt-mankowski/python/ch-2.py b/challenge-070/walt-mankowski/python/ch-2.py
new file mode 100644
index 0000000000..be0f8ac2d3
--- /dev/null
+++ b/challenge-070/walt-mankowski/python/ch-2.py
@@ -0,0 +1,14 @@
+from sys import argv
+
+n = int(argv[1])
+S = [0, 1, 3, 2]
+
+def gray_code(n, S):
+    fmt = f"0{n-1}b"
+    S2 = [eval(f"0b1{x}") for x in [f"{x:{fmt}}" for x in S][::-1]]
+    return S + S2
+
+for i in range(3, n+1):
+    S = gray_code(i, S)
+
+print(S)