demonstration of how fast numpy operations can be used to speed up part 2 of aoc

aoc23-demo05.py

import numpy as np
from more_itertools import chunked

map = [
    [50, 98, 2],
    [52, 50, 48]
]

def apply_map(map, srcs):
    for dst_start, src_start, range_len in map:
        srcs[np.where(np.logical_and(
            src_start <= srcs,
            srcs < src_start + range_len
        ))] += dst_start - src_start

if __name__ == '__main__':
    seeds = [
        79, 14,
        55, 13
    ]
    for seed_start, seed_range in chunked(seeds, 2):
        locs = np.fromiter(range(seed_start, seed_start + seed_range), int)
        apply_map(map, locs)
        print(f"{seed_start=} {seed_range=} -> ", locs)