CODEWITHSUNDEEP
pythonunpack
Hanfei Sun
Hanfei Sun

Reputation: 47071

Unpack the first two elements in list/tuple

Is there a way in Python to do like this:

a, b, = 1, 3, 4, 5

and then:

>>> a
1
>>> b
3

The above code doesn't work as it will throw

ValueError: too many values to unpack

Upvotes: 93

Views: 94159

Answers (5)

Gustav Larsson
Gustav Larsson

Reputation: 8497

Just to add to Nolen's answer, in Python 3, you can also unpack the rest, like this:

>>> a, b, *rest = 1, 2, 3, 4, 5, 6, 7
>>> a
1
>>> rest
[3, 4, 5, 6, 7]

(this does not work in Python 2)

Upvotes: 168

Ned Batchelder
Ned Batchelder

Reputation: 375892

There is no way to do it with the literals that you've shown. But you can slice to get the effect you want:

a, b = [1, 3, 4, 5, 6][:2]

To get the first two values of a list:

a, b = my_list[:2]

Upvotes: 74

jamylak
jamylak

Reputation: 133684

On Python 3 you can do the following:

>>> a, b, *_ = 1, 3, 4, 5
>>> a
1
>>> b
3

_ is just a place holder for values you don't need

Upvotes: 44

Levon
Levon

Reputation: 143122

Or in Python 3.x you could do this:

  a, *b = 1, 3, 4

giving you:

In [15]: a
Out[15]: 1

In [16]: b
Out[16]: [3, 4]

It would avoid the exception, though you would have to parse b. This assume that you only want to have two variables on the left of the =, otherwise you could use

a, b, *ignore = ....

with v3.x

Upvotes: 4

Nolen Royalty
Nolen Royalty

Reputation: 18653

You could use _ to represent variables you wanted to "throw away"

>>> a, b, _ = 1, 3, 4
>>> a
1
>>> b
3

Upvotes: 5

Related Questions