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] return [5, 2, 8, 7]（或相应的索引 - 顺序并不重要）。

显然有一个简单的解决方案，但是，它太慢了，因为我想大规模地为具有几千个元素的数组做这个。有什么建议？提前谢谢。

Answer 1

为什么太慢？

从输入坐标[a，b]返回列表[[a-1，b]，[a + 1，b]，[a，b-1]，[a，b + 1]]，避免网格外的坐标。