python - 使用 numpy 填充数组

Question

在我的程序中，我有一个 numpy 数组并对其进行了一些卷积过滤。我正在寻找一些方法来使用 numpy 轻松地进行数组填充(然后为输出取消填充)以避免边界检查。我知道 scipy 可以做卷积，但我有理由自己做。 gnuplot.py 用于输出。

def touch(field, coords, value):
    field[coords[0], coords[1]] = value
    if coords[0] - 1 > 0:
        field[coords[0] - 1, coords[1]] = value / 2
    if coords[1] - 1 > 0:
        field[coords[0], coords[1] - 1] = value / 2
    if coords[0] < field.shape[0] - 1:
        field[coords[0] + 1, coords[1]] = value / 2
    if coords[1] < field.shape[1] - 1:
        field[coords[0], coords[1] + 1] = value / 2

Answer 1

有一个 pad 模块计划包含在 Numpy 1.7.0 中 – 参见 this ticket .现在，只需下载它并使用它的 with_constant 函数。

取消填充就像 field[1:-1, 1:-1] 一样简单。