assign unknown number of return values to single variable in python

Question

My function is similar to following

def function(n):
    if n == 1:
        return [1, 2, 3]
    if n == 2:
        return