Python:将数组拆分为变量

我有一个返回列表的函数.调用者不知道该列表的长度.

def get_array():
    ret = [1, 2]
    return ret

var1, var2 = get_array()

这套var1 = 1和var2 = 2.这适用于所有列表,除非长度小于2.例如,如果列表的长度为1:

 def get_array():
    ret = [1]
    return ret

 var1 = get_array()

这设置var1为[1],但我想var1成为1.是否有一种更清洁的方法来打开这个阵列而不必检查它的长度？

如果你正在使用python3,你可以使用splatted unpacking语法来帮助你:

In [113]: a,*b = [1]

In [114]: a
Out[114]: 1

In [115]: b
Out[115]: []

基本上,这可以让你解压缩get_array()[1:]到b,这将是空的,如果get_array()返回大小为1的列表