Как определить соседей в двумерном списке?

Question

[aab137]

Есть квадратная матрица, например,

[[4,3,5],
 [2,8,6],
 [0,7,9]]

Нужно получить список соседей для каждой "клетки", например, для данной матрицы он будет таким:

[[2,8,3], [4,2,8,6,5], [3,8,6], [4,3,8,7,0], [4,3,5,6,9,7,0,2], и т.д.]

Хотелось бы сделать это максимально быстрым способом

Answer 1

[Сергей Горностаев]

def neighbors(x):
    def in_bounds(i, j, x):
        return 0 <= i < len(x) and 0 <= j < len(x[0])

    return [[x[i + di][j + dj] for di in (-1, 0, 1)
                               for dj in (-1, 0, 1)
                               if in_bounds(i + di, j + dj, x) and (di != 0 or dj != 0)]
            for i in range(len(x))
            for j in range(len(x[0]))]

Answer 2

[xmoonlight]

Делайте линейный массив (бит-блок) и вычисляйте с помощью целочисленного деления и остатка с нужным приращением (+/- 1).

Comments

[aab137]

Я, будучи чайником, немного не понял, что Вы написали.

[xmoonlight]

aab137, [4,3,5,2,8,6,0,7,9]

[aab137]

xmoonlight, не понял, ри чем тут остаток и деление

[aab137]

а, все понял