自定义中值滤波中位数计算函数出现IndexError错误求助

我帮你排查一下这个索引越界的问题，其实是两个关键细节出错了，咱们一步步拆解：

问题1：循环范围少遍历了一行/一列 #

你写的for i in range(gridSize-1)和for j in range(gridSize-1)有问题——Python的range()是左闭右开的，比如当你用3x3的滤波核（gridSize=3）时，range(2)只会循环0和1，漏掉了索引2对应的那一行/列，导致medianList的元素数量比预期少。更糟的是，在处理图像边缘时，可能会出现列表为空的情况，最后取中位数索引自然就越界了。

问题2：边界处理用了break而非continue #

当检测到坐标越界时，你用break直接终止了当前的j循环，这会导致同一i值下的后续j元素都被跳过，进一步减少medianList的元素数量，甚至在极端情况（比如图像角落的像素）下，列表会直接为空，触发IndexError。

修正后的代码 #

这里是调整后的函数，我标注了关键修改点：

def CalcMedian(Image, x, y, gridSize):
    # x and y are the coordinates of the current pixel being processed
    medianList = []
    row, col = Image.shape
    k = int(gridSize / 2)
    # 修正：遍历整个滤波核的所有元素，而不是gridSize-1
    for i in range(gridSize):
        for j in range(gridSize):
            current_row = i + x - k
            current_col = j + y - k
            # 修正：仅将合法范围内的像素加入列表，越界则跳过而非终止循环
            if 0 <= current_row < row and 0 <= current_col < col:
                medianList.append(Image[current_row, current_col])
            else:
                continue
    medianList.sort()
    length = len(medianList)
    if length % 2 != 0:
        return float(medianList[length // 2])
    # 简化偶数长度的中位数计算，逻辑和原代码一致但更直观
    return float((medianList[length//2 - 1] + medianList[length//2]) / 2.0)

额外说明 #

• 现在即使处理图像边缘的像素，越界的像素会被直接跳过，合法的像素依然会被收集到medianList中，不会出现列表为空的情况（至少当前像素(x,y)本身是合法的，会被加入列表）。
• 偶数长度的中位数计算我调整了索引写法，和原代码逻辑完全一致，但更易读。

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