Chris Smith пре 6 година
родитељ
комит
97c4a095d6
1 измењених фајлова са 2 додато и 2 уклоњено
  1. 2
    2
      01.py

+ 2
- 2
01.py Прегледај датотеку

8
 import numpy as np
8
 import numpy as np
9
 with open('data/01.txt', 'r') as file:
9
 with open('data/01.txt', 'r') as file:
10
     captcha = list(file.readline().strip())
10
     captcha = list(file.readline().strip())
11
-    print('Part one: %s' % sum(int(i) for i, j in zip(captcha, np.roll(captcha, 1).tolist()) if i == j))
12
-    print('Part two: %s' % sum(int(i) for i, j in zip(captcha, np.roll(captcha, len(captcha)//2).tolist()) if i == j))
11
+for part, roll in enumerate([1, len(captcha)//2]):
12
+    print('Part %s: %s' % (part + 1, sum(int(i) for i, j in zip(captcha, np.roll(captcha, roll)) if i == j)))

Loading…
Откажи
Сачувај