#!/bin/python3 import re import sys from pprint import pprint input_f = sys.argv[1] count = 0 arr = [] with open(input_f) as file: for line in file: arr.append(list(line.lstrip().rstrip())) pprint(arr) X=len(arr[0]) Y=len(arr) neighbors = lambda x, y : [(x2, y2) for x2 in range(x-1, x+2) for y2 in range(y-1, y+2) if (-1 < x <= X and -1 < y <= Y and (x != x2 or y != y2) and (0 <= x2 <= X) and (0 <= y2 <= Y))] pprint(neighbors(4,2))