Как распаковать кортеж длины n в m

В Python 3 я могу сделать следующее (см. также PEP3132 о расширенной распаковке итерабельных файлов):

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

Что я могу сделать, чтобы добиться такого же элегантного результата в Python 2.x?

Я знаю, что могу использовать одноэлементный доступ и операции нарезки, но мне интересно, есть ли более питонический способ. Мой код на данный момент:

a, b = (1, 2, 3)[0], (1, 2, 3)[1:]
# a = 1; b = (2, 3)
moooeeeep
Редактировал вопрос 22-го сентября 2014 в 1:11
Программирование
python iterable-unpacking
24-го апреля 2012 в 2:08
19 просмотров
Комментарии к вопросу (4)
Решение

Я узнал, что соответствующие PEP3132 дает некоторые примеры для Python 2.X, а также:

многие алгоритмы требуют расщепления последовательности в "первый, остальные" в паре:

В первый, остальные = сл[0], далее[1:]

[...]

также, если правое значение не список, а итератор, он должен быть преобразован в список, прежде чем он сможет сделать нарезку; чтобы избежать этого временного списка, приходится прибегать к

Это = ИТЭР(сл) первый = он.далее() остальные = список ()

Другие подходы приведенный в ответах на этот вопрос:

Список Аргументов Функции Распаковки Подход

требует дополнительного определения функции/вызов:

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

Интересно, почему он реализуется в распаковке функции список аргументов, но не для нормальной распаковки кортежа.

Подход Генератор

Авторы. Также требует пользовательской реализации функции. Это немного более гибким относительно числа переменных first.

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)

Наиболее подходящие для Python, вероятно, будут те, которые упомянуты в выше Пеп, я думаю?

Community
Редактировал ответ 23-го мая 2017 в 12:02
Комментарии (3)

Я могу ошибаться, но насколько я знаю.

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

это просто синтаксический сахар для нарезки и индексирования кортежей. Я нахожу его полезным, но не очень явным.

Комментарии (4)

Я'ве получил этой маленькой, но удобной функции:

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)
georg
Редактировал ответ 24-го апреля 2012 в 3:13
Комментарии (5)

Я не думаю, что есть лучший способ, чем тот, который вы опубликовали, но вот альтернативный вариант с использованием iter.

>>> x = (1,2,3)
>>> i = iter(x)
>>> a,b = next(i), tuple(i)
>>> a
1
>>> b
(2, 3)
Комментарии (0)

Не уверен в контексте, но как насчет .pop(0)?

Я вижу, что в вашем примере есть кортежи, но если вы хотите делать то, что вы делаете, списки были бы более подходящими, я думаю? (Если только нет какой-то веской причины для того, чтобы они были неизменяемыми, не указанной в вопросе).

b = [1,2,3]
a = b.pop(0)
Комментарии (0)