skrd/advent-of-code
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Now using regex for data parsing

Browse Source
This commit is contained in:
Daniel Cortés
2020-12-07 07:31:00 -03:00
parent 675f82e72d
commit 6659174e41
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
2020/day_7/handy_haversacks.py
View File

@@ -1,14 +1,14 @@
import re
def process_rules(data): def process_rules(data):
rules = {} rules = {}
for line in data: for line in data:
line = line.replace(' bags', '').replace(' bag', '').replace('.', '') color = re.search("(\w+ \w+) bags contain", line).group(1)
color, rule = [r.strip() for r in line.split('contain ')] rule = re.findall("(\d+) (\w+ \w+)", line)
rule = rule.split(', ')
rules[color] = [] rules[color] = []
for bag in rule: for bag in rule:
if bag == 'no other': break rules[color].append((bag[1], int(bag[0])))
rules[color].append((bag[2:], int(bag[0])))
return rules return rules