Head and tail in one line

by

Under Python 3.x, you can do this nicely:

>>> head, *tail = [1, 1, 2, 3, 5, 8, 13, 21, 34, 55]
>>> head
1
>>> tail
[1, 2, 3, 5, 8, 13, 21, 34, 55]

A new feature in 3.x is to use the * operator in unpacking, to mean any extra values. It is described in PEP 3132 – Extended Iterable Unpacking. This also has the advantage of working on any iterable, not just sequences.

It’s also really readable.

As described in the PEP, if you want to do the equivalent under 2.x (without potentially making a temporary list), you have to do this:

it = iter(iterable)
head, tail = next(it), list(it)

As noted in the comments, this also provides an opportunity to get a default value for head rather than throwing an exception. If you want this behaviour, next() takes an optional second argument with a default value, so next(it, None) would give you None if there was no head element.

Naturally, if you are working on a list, the easiest way without the 3.x syntax is:

head, tail = seq[0], seq[1:]

More Related Contents:

  1. Printing the number from a list and its square on the same line in python3 [closed]
  2. Python3 – create global variable(list) from function
  3. Storing variables in a list when not initially assigned in python
  4. How to reverse a list?
  5. How does assignment work with list slices?
  6. Count frequency of words in a list and sort by frequency
  7. How to avoid “RuntimeError: dictionary changed size during iteration” error?
  8. How to find all positions of the maximum value in a list?
  9. In Python, what is the difference between “.append()” and “+= []”?
  10. a mutable type inside an immutable container
  11. Different ways of clearing lists
  12. How to get item’s position in a list?
  13. Add list to set
  14. How to find the min/max value of a common key in a list of dicts?
  15. str.startswith with a list of strings to test for
  16. How to split strings inside a list by whitespace characters
  17. Writing Python lists to columns in csv
  18. Why python’s list slicing doesn’t produce index out of bound error? [duplicate]
  19. Index all *except* one item in python
  20. How to convert a string with comma-delimited items to a list in Python?
  21. creating a defaultlist in python
  22. Converting list to *args when calling function [duplicate]
  23. Find the min, max value in a list of tuples
  24. Mapping dictionary value to list
  25. Creating sublists [duplicate]
  26. Can’t sort my list because it is NoneType? Simple Python [duplicate]
  27. How to sort a list of dictionaries by a value of the dictionary in Python?
  28. Why Python `Memory Error` with list `append()` lots of RAM left
  29. How to change variables fed into a for loop in list form
  30. list extend() to index, inserting list elements not only to the end

Leave a Comment