From 32fe21038a75a3f6da911c92247e422885cb521b Mon Sep 17 00:00:00 2001
From: Thomas Voss <thomasvoss@live.com>
Date: Wed, 1 Dec 2021 08:55:41 +0100
Subject: Make use of the magic zip function
---
2021/01/puzzles.py | 5 +----
1 file changed, 1 insertion(+), 4 deletions(-)
(limited to '2021/01')
diff --git a/2021/01/puzzles.py b/2021/01/puzzles.py
index 0a894ca..d9a704b 100644
--- a/2021/01/puzzles.py
+++ b/2021/01/puzzles.py
@@ -4,10 +4,7 @@ def main() -> None:
with open("input", "r") as f:
nums = [int(n) for n in f.readlines()]
- # START PART 2
- nums = [nums[i] + nums[i - 1] + nums[i - 2] for i in range(2, len(nums))]
- # END PART 2
- print(sum(nums[i] > nums[i - 1] for i in range(1, len(nums))))
+ print(sum(n < m for n, m in zip(nums, nums[DELTA:])))
if __name__ == "__main__":
main()
--
cgit v1.2.3