请问这个代码该怎么修改呢？

Question

当input1 = [1, 2, 3, 4, 5]
input2 = [2, 2]
head_tail_compare(input1)，
这段代码输出结果是(False, 5)，
请问有什么办法让它变成：
False
5
这样吗

def head_tail_compare(lst):
    """
    A function that compares the first element and the last element in the 
    input list.
    ---
    Parameters:
    lst: a list of numbers
    ---
    Print True if they are the same, False if not
    Return the larger element

    >>> input1 = [1, 2, 3, 4, 5]
    >>> input2 = [2, 2]
    >>> head_tail_compare(input1)
    False
    5
    >>> head_tail_compare(input2)
    True
    2
    """
    # YOUR CODE GOES HERE #
    if lst[0]==lst[-1]:
        return True
    if lst[0]!=lst[-1] and lst[0]>lst[-1]:
        return False,lst[0]
    if lst[0]!=lst[-1] and lst[0]<lst[-1]:
        return False,lst[-1]
    return

Answer 1

用一个变量接受在遍历输出即可

 
def head_tail_compare(lst):

    # YOUR CODE GOES HERE #
    if lst[0]==lst[-1]:
        return True
    if lst[0]!=lst[-1] and lst[0]>lst[-1]:
        return False,lst[0]
    if lst[0]!=lst[-1] and lst[0]<lst[-1]:
        return False,lst[-1]
    return
a=head_tail_compare( [1, 2, 3, 4, 5])
for i in a:
    print(i)