如何迭代矩阵数组，以计算矩阵中特定元素周围的相似元素的数量

def count_similar_elements(matrix, row, col, threshold):
    similar_count = 0
    target_element = matrix[row][col]
    rows = len(matrix)
    cols = len(matrix[0])

    for i in range(row-1, row+2):
        for j in range(col-1, col+2):
            if i >= 0 and i < rows and j >= 0 and j < cols:
                if abs(matrix[i][j] - target_element) < threshold:
                    similar_count += 1

    return similar_count

# 示例矩阵数组
matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 计算矩阵中元素5周围相似元素的数量，假设相似元素的阈值为2
similar_count = count_similar_elements(matrix, 1, 1, 2)
print(similar_count)