Проект "Марсохід"

У файлі записана "карта" Марсу:

Ввівши координати місця розташування марсоходу, визначити, що він "бачить" навколо. 

Зчитування поля з файлу

f=open("mars.txt")
terrain=[]
for ryadok in f.readlines():
    terrain.append(ryadok.split())

Друк поля

for i in range(6):
    for j in range(6):
        print("%10s"% terrain[i][j], end="")
    print()

Функція визначення сусідів клітинки

def view(pole,a,b):
    if a==0 and b==0:
        susidy=[[a,b+1],[a+1,b],[a+1,b+1]]
    elif a==9 and b==9:
        susidy=[[a-1,b-1],[a-1,b],[a,b-1]]
    elif a==0 and b==9:
        susidy=[[a,b-1],[a+1,b-1],[a+1,b]]
    elif a==9 and b==0:
        susidy=[[a-1,b],[a-1,b+1],[a,b+1]]
    elif a==0:
        susidy=[[a,b-1],[a,b+1],[a+1,b-1],[a+1,b],[a+1,b+1]]
    elif b==0:
        susidy=[[a-1,b],[a-1,b+1],[a,b+1],[a+1,b],[a+1,b+1]]
    elif a==9:
        susidy=[[a-1,b-1],[a-1,b],[a-1,b+1],[a,b-1],[a,b+1]]
    elif b==9:
        susidy=[[a-1,b-1],[a-1,b],[a,b-1],[a+1,b-1],[a+1,b]]
    else:
        susidy=[[a-1,b-1],[a-1,b],[a-1,b+1],[a,b-1],[a,b+1],[a+1,b-1],[a+1,b],[a+1,b+1]]
    k=0
    for element in susidy:
        if pole[element[0]][element[1]]==1:
            k+=1
    return(k)

Показати код/Сховати код^

Передавання інформації про вид з клітинки, координати якої введено з клавіатури

a=input().split()
x=int(a[0])
y=int(a[1])
print(terrain[x][y])
print("Я бачу навколо",view(terrain,x,y))

Що насправді відбувається на Марсі ;)

https://www.e-olymp.com/uk/problems/2671

[['Озеро', 'Гора', 0, 0, 'Озеро', 'Озеро'], ['Озеро', 'Марсіанин', 'Гора', 'Долина', 'Марсіанин', 'Гора'], [0, 0, 'Марсіанин', 'Озеро', 'Озеро', 0], [0, 0, 'Гора', 'Гора', 'Озеро', 0], ['Озеро', 0, 'Марсіанин', 'Марсіанин', 'Озеро', 0], ['Озеро', 'Долина', 'Марсіанин', 0, 'Озеро', 'Гора']]