def box_range(number):
    """
    :param number: input the row or column number
    :return: a list of row or column number within the same box
    """
    if number in (1, 2, 3):
        return [1, 2, 3]
    elif number in (4, 5, 6):
        return [4, 5, 6]
    elif number in (7, 8, 9):
        return [7, 8, 9]