如何为NumPy ndarray添加缓冲带以满足栅格数据均等切分需求

你需要的缓冲带填充功能可以直接用NumPy内置的np.pad()函数实现，无需额外封装自定义函数。

示例匹配实现 #

对应你给出的3x3数组添加1层0值缓冲的需求，代码如下：

import numpy as np

a = np.array([
    [1,4,5],
    [4,5,5],
    [1,2,2]
])

# pad_width：每个维度前后的填充宽度，mode='constant'表示用固定值填充
a_with_buffer = np.pad(a, pad_width=1, mode='constant', constant_values=0)

输出结果完全符合你的预期：

[[0 0 0 0 0]
 [0 1 4 5 0]
 [0 4 5 5 0]
 [0 1 2 2 0]
 [0 0 0 0 0]]

自定义填充配置 #

• 若要使用NaN作为填充值，修改constant_values参数即可，注意需要先将数组转为浮点型：

  a = a.astype(np.float32)
  a_with_buffer_nan = np.pad(a, pad_width=1, mode='constant', constant_values=np.nan)
  

• 若仅需要在单侧（如下方、右侧）填充适配瓦片尺寸，可单独指定各边填充宽度：
  以你提到的1490宽度栅格适配250x250瓦片为例，计算填充量的代码如下：

  tile_size = 250
  # 假设读取的单波段栅格数组shape为(行数, 列数)
  rows, cols = rasterio_arr.shape
  # 计算需要补充的行列数，取模避免刚好整除时产生多余填充
  need_pad_rows = (tile_size - rows % tile_size) % tile_size
  need_pad_cols = (tile_size - cols % tile_size) % tile_size
  # 仅在下方、右侧填充，适合栅格瓦片切分场景
  padded_arr = np.pad(
      rasterio_arr,
      pad_width=((0, need_pad_rows), (0, need_pad_cols)),
      mode='constant',
      constant_values=0
  )
  

栅格存储注意事项 #

如果填充后需要用rasterio写回.tif文件，记得同步更新栅格的width、height属性，以及仿射变换参数transform，避免地理坐标偏移。

内容的提问来源于stack exchange，提问作者Sytze