#!/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))] for idx,i in enumerate(arr): for jdx,j in enumerate(i): tmp = arr[idx][jdx] if tmp.isdigit() == False and tmp != '.': print(tmp) print(idx,jdx) n = neighbors(4,2) pprint(n) for i in n: print(i) print(arr[i[0]][i[1]])