如何在二维数组中从中心扩展数字的半径

def expand_array_radius(array, radius):
    rows = len(array)
    cols = len(array[0])

    center_row = (rows - 1) // 2
    center_col = (cols - 1) // 2

    row = center_row
    col = center_col

    # Expand the radius
    for r in range(1, radius + 1):
        # Expand to the right
        for i in range(r):
            col += 1
            array[row][col] = array[row][col-1] + 1

        # Expand downwards
        for i in range(r):
            row += 1
            array[row][col] = array[row-1][col] + 1

        # Expand to the left
        for i in range(r+1):
            col -= 1
            array[row][col] = array[row][col+1] + 1

        # Expand upwards
        for i in range(r+1):
            row -= 1
            array[row][col] = array[row+1][col] + 1

    return array

# Example usage
array = [[0, 0, 0],
         [0, 0, 0],
         [0, 0, 0]]

radius = 2
expanded_array = expand_array_radius(array, radius)

print(expanded_array)