Как вывести соседние элементы в матрице?

Question

[kyurem_bog]

Дана матрица. Напечатать те ее элементы, все соседи которых по горизонтали, вертикали и диагонали: а) больше этого элемента, б) меньше этого элемента. Как мне указать в условии что надо выводить только те которые меньше или больше элемента

import random
def start():
        print("Дана матрица")        
        def creatArray():
            r = 0
            print('Input first index matrix: ')
            x = int(input())
            print('Input second index matrix: ')
            y = int(input())
            array = []
            for i in range(x):
                array.append([])
                for j in range(y):
                    array[i].append(random.randint(0,100))
                    r += 1  
            return array
        g= creatArray()
        def printMatrix(g): 
           for row in g: 
              for x in row: 
                  print ( "{:3d}".format(x), end = "" ) 
              print ()
        printMatrix (g)
        print("Выбирите a или b")
        a= input()        
        if a == "a":
                def in_bounds(i, j, g):
                        return 0 <= i < len(g) and 0 <= j < len(g[0])
                def neighbors(g):
                        return [[g[i + di][j + dj] for di in (-1, 0, 1) for dj in (-1, 0, 1) if in_bounds(i + di, j + dj, g) and (di != 0 or dj != 0)]
                                for i in range(len(g)) for j in range(len(g[0]))]
                print(neighbors(g))
                #именно эта часть,  выводит соседние элементы каждого элемента.
                #но нужно сделать так чтобы выводила только те которые больше либо меньше этого элемента
 
        elif a == "b":
            s = 0 
            for row in matrix: 
               s += sum(row) #просто заглушка
            print (s)
        else:
            print('Other')
start()
while True:
    flag = input('Ещё раз? [да/нет]: ')
 
    if flag == 'да':
            start()
    else:
        break

Comments

[o5a]

Судя по условиям задачи требуется не "чтобы выводила только те которые больше либо меньше этого элемента", а выводить только элементы матрицы, у которой все соседи больше/меньше его самого.

Т.е. можно например оставить эту функцию, возвращающую всех соседей (только убрать условие больше/меньше). Затем пройтись по всем элементам матрицы, и проверить, что для каждого элемента его список neighboors больше/меньше самого элемента.

[kyurem_bog]

o5a, ну вот я просто не пойму как это написать, вот и спрашиваю

[o5a]

kyurem_bog, я сначала думал, что neigboors просто возвращает соседей, типа такого

def neighbors(i, j):
    return [g[i + di][j + dj] for di in (-1, 0, 1) for dj in (-1, 0, 1) if (di != 0 or dj != 0)]

Тогда можно было циклом пройтись по матрице и для каждого (i,j) возвращать список соседей, проверять, что все из списка больше/меньше самого элемента, и тогда выводить элемент.

Но раз функция сразу проходит по всей матрице можно и так, добавить эту проверку внутрь функции

def neighbors_min(g):
    return [g[i][j] if min([g[i + di][j + dj] for di in (-1, 0, 1) for dj in (-1, 0, 1) if (di != 0 or dj != 0)]) > g[i][j]
        for i in range(len(g)) for j in range(len(g[0]))]

Т.е. выводить те элементы матрицы, у которых минимальный из соседей больше самого элемента. Это как раз будет условие "а". Аналогично для "б".

[kyurem_bog]

o5a, пишет вот такое
File "main.py", line 31
for i in range(len(g)) for j in range(len(g[0]))]
^
SyntaxError: invalid syntax

[o5a]

kyurem_bog, просто условие if надо было в конец вынести, я сразу не глянул. Плюс надо еще добавлять условие выхода за границы матрицы. Примерно так

def neighbors(g):
    return [g[i][j] for i in range(len(g)) for j in range(len(g[0]))
        if max([g[i + di][j + dj] for di in (-1, 0, 1) for dj in (-1, 0, 1) if (di != 0 or dj != 0) and (i + di < len(g)) and (j + dj < len(g[0]))]) < g[i][j]]

Получилось громоздко из-за того, что все внутрь запихали, но идея думаю понятна, так должно работать.

[Алан Гибизов]

o5a, а вы коварны! Решаете за студентов задачи, тем самым снижаете уровень будущих потенциальных конкурентов. Хитро!

[kyurem_bog]

o5a, Вот если честно, чет ничего не выходит. В данный момент оно выводит просто больший элемент будто.