30 lines
660 B
Python
30 lines
660 B
Python
|
#!/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))
|
||
|
|
|