python - 在 4-D 数组中查找相邻值

Question

我喜欢在 4-D 数组中找到每个元素的邻居。我曾尝试使用此答案中给出的方法，但在某些方面它似乎对我不起作用。

作为一个例子，我正在尝试使用该答案底部给出的简单方法：

p = np.r_[0, 0]
shape = (6, 6)
x = np.zeros(shape, int)
x[tuple(p)] = 2
idx = tuple(slice(pp - 1, pp + 2) for pp in p)
print(x[idx])

我期待得到

[[2 0]
 [0 0]]

相反，我得到一个空数组。谁能指出我在这里缺少的东西？谢谢

标签： pythonnumpy