from pathlib import Path import sys from collections import defaultdict from itertools import permutations pids = defaultdict(list) for p in Path(sys.argv[1]).glob("*.jpg"): pids[p.stem[:7]].append(p.stem) data = [] for p in pids: data.extend(list(permutations(pids[p], 2)))