gpt4 book ai didi

python - 如何在返回值中获取列表的一部分

转载 作者:太空宇宙 更新时间:2023-11-04 07:59:01 26 4
gpt4 key购买 nike

我想在一个方法中返回多个值,包括一个列表。

以下工作正常:

def foo():
a = 42
b = [1,2]
return a, b

aa, b_list = foo()
aa, (b1, b2) = foo()

但我不知道如何得到没有第一个元素的列表:

def bar():
a = 42
b = [1,2,3]
return a, b

aa, (_, b_list) = bar()
print(b_list) # Expecting [2,3]
# ValueError: too many values to unpack (expected 2)

是的,我可以获取列表并删除第一个元素,但是是否有单行代码?

这个函数应该是通用的,比如应该支持两种场景:

aa, whole_list = foo()
aa, first_elem, rest_of_the_list = foo()

最佳答案

是的,有。您可以对 b_list 使用解包:

aa, (_, *b_list) = bar()
  • 请注意,这仅适用于 python 3。

关于python - 如何在返回值中获取列表的一部分,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43794904/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com