Быстрый метод Python для получения соседних элементов в 2D сетке

Question

Есть ли способ где-нибудь в пакете Python, который возвращает элементы и/или индексы элемента в 2-й сетке. Например. если мы имеем:

[[1, 2, 3, 4], 

[5, 6, 7, 8], 

[7, 8, 9, 0]] 

..и мы даем метод индекс [0,1] он должен вернуть [1, 6, 3] (если он может вернуться [[0,0], [1,1], [0,2]], что было бы еще лучше) и придав ему [1,1] бы вернуть [5, 2, 8, 7] (или соответствующий indexes- порядок не важен).

Очевидно, что это простое решение, но оно слишком медленное, так как я хочу сделать это в больших масштабах для массивов с несколькими тысячами элементов. Какие-либо предложения? Заранее спасибо.

Answer 1

Почему это слишком медленно?

Из входной координаты [a, b] возвращают список [[a-1, b], [a + 1, b], [a, b-1], [a, b + 1]], избегая координат вне вашей сетки.