python函數(shù)返回值只取其中一個怎么實現(xiàn)

def test():
    return 1,2
a, b = test()
1 2
a, _ = test()
1   
 
# 說明 a, _ = test() 中 返回的是一個int類型 后續(xù)可直接運算
# a, b = test()  返回值是一個元組, 直接運算報錯

import time
import random
import string
import functools
#問題：被裝飾的函數(shù)有返回值
li = [random.choice(string.ascii_letters) for i in range(100)]
def timeit(fun):
    @functools.wraps(fun)  #為了保留被裝飾函數(shù)的函數(shù)名和幫助文檔信息
    def wrapper(*args,**kwargs):
        """這是一個wrapper函數(shù)"""
        start_time = time.time()
        res = fun(*args,**kwargs)
        end_time = time.time()
        print('運行時間為:%.6f' %(end_time - start_time))
        return res
    return wrapper
@timeit
def fun_list(n):
    """這是fun_list函數(shù)"""
    return [i * 2 for i in range(n)]
@timeit
def fun_map(n):
    """這是fun_map函數(shù)"""
    return list(map(lambda x:x*2,range(n)))
print(fun_list(5000))
print(fun_map(5000))