如何解读长度为n到m的元组

时间:2012-04-24 14:08:42

标签: python iterable-unpacking

在Python 3中,我可以执行以下操作(另请参阅有关扩展可迭代解包的PEP3132):

a, *b = (1, 2, 3)
# a = 1; b = (2, 3)

我可以做些什么来在Python 2.x中实现同样优雅?


我知道我可以使用单个元素访问和切片操作,但我想知道是否有更多的 pythonic 方式。到目前为止我的代码:

a, b = (1, 2, 3)[0], (1, 2, 3)[1:]
# a = 1; b = (2, 3)

5 个答案:

答案 0 :(得分:29)

我发现相关的PEP3132也为Python 2.x提供了一些示例:

  

许多算法都需要在“first,rest”对中拆分序列:

first, rest = seq[0], seq[1:]
     

[...]

     

此外,如果右侧值不是列表,而是可迭代的,则必须先将其转换为列表才能进行切片;为了避免创建这个临时列表,必须诉诸

it = iter(seq)
first = it.next()
rest = list(it)

在这个问题的答案中给出的其他方法:

功能参数列表拆包方法

需要额外的函数定义/调用:

def unpack(first, *rest): 
  return first, rest
first, rest = unpack( *seq )

我想知道为什么它在解包函数参数列表中实现,但不适用于正常的元组解包。

生成方法

Credits。还需要自定义函数实现。关于第一个变量的数量,它会更加灵活。

def unpack_nfirst(seq, nfirst):
  it = iter(seq)
  for x in xrange(nfirst):
    yield next(it, None)
  yield tuple(it)
first, rest = unpack_nfirst(seq, 1)

最疯狂的可能就是上面PEP中提到的那些,我猜?

答案 1 :(得分:8)

我有这个方便的小功能:

def just(n, seq):
    it = iter(seq)
    for _ in range(n - 1):
        yield next(it, None)
    yield tuple(it)

例如:

a, b, c = just(3, range(5))
print a, b, c
## 0 1 (2, 3, 4)

也可以使用较少的参数:

a, b, c = just(3, ['X', 'Y'])
print a, b, c
## X Y ()

在回复评论时,您还可以定义:

def take2(a, *rest): return a, rest
def take3(a, b, *rest): return a, b, rest
def take4(a, b, c, *rest): return a, b, rest
... etc

并像这样使用它:

p = (1,2,3)
a, b = take2(*p)
print a, b
## 1 (2, 3)

答案 2 :(得分:6)

我可能错了,但据我所知

a, *b = (1, 2, 3)

只是用于切片和索引元组的语法糖。我发现它很有用,但不是很明确。

答案 3 :(得分:4)

我认为没有比你发布的方式更好的方法,但这里有一个使用iter的替代方案

>>> x = (1,2,3)
>>> i = iter(x)
>>> a,b = next(i), tuple(i)
>>> a
1
>>> b
(2, 3)

答案 4 :(得分:2)

不确定上下文,但是.pop(0)呢?

我看到你的例子中有元组,但如果你想做你做的那些事情,我认为列表会更合适吗? (除非有一些很好的理由让它们在问题中没有提到。)

b = [1,2,3]
a = b.pop(0)