写函数，检查传入长度，

def check_len(a):
    if len(a) > 2:
        return [a[0], a[1]]
    return a