快速Python方法在2D網格中獲取相鄰元素

Question

在Python包中是否存在返回2d網格中元素和/或索引的方法。例如。如果我們有：

[[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]]，避免網格外的座標。