#!/usr/bin/env python3 from re import findall from functools import reduce # filename = "in/day05.ref" filename = "in/day05.pzl" m = [ [i for i in line.rstrip('\n')[1::4] ] for line in open(filename) if line.find('[') != -1] # print('m:', m) mt = [[m[col_n][row_n] for col_n,_ in enumerate(m) if m[col_n][row_n]!=' '][::-1] for row_n,_ in enumerate(m[0])] # print('mt:', mt) a = [ [int(val) if i==0 else int(val)-1 for i,val in enumerate(findall('\d+', line))] for line in open(filename) if line[0] == 'm'] # print('a:', a) b1 = reduce(lambda state, action: [row[:-action[0]] if row_n==action[1] else row+state[action[1]][-action[0]:][::-1] if row_n==action[2] else row for row_n, row in enumerate(state)], a, mt) # print('b1:', b1) res1 = ''.join([i[-1] for i in b1]) print('res1:', res1) b2 = reduce(lambda state, action: [row[:-action[0]] if row_n==action[1] else row+state[action[1]][-action[0]:] if row_n==action[2] else row for row_n, row in enumerate(state)], a, mt) # print('b2:', b2) res2 = ''.join([i[-1] for i in b2]) print('res2:', res2)