Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

What is the pythonic way to detect the last element in a 'for' loop?

I'd like to know the best way (more compact and "pythonic" way) to do a special treatment for the last element in a for loop. There is a piece of code that should be called only between elements, being suppressed in the last one.

Here is how I currently do it:

for i, data in enumerate(data_list):     code_that_is_done_for_every_element     if i != len(data_list) - 1:         code_that_is_done_between_elements 

Is there any better way?

Note: I don't want to make it with hacks such as using reduce. ;)

like image 936
e.tadeu Avatar asked Oct 27 '09 11:10

e.tadeu


People also ask

How do I find the last element in a list?

Any element in list can be accessed using zero based index. If index is a negative number, count of index starts from end. As we want last element in list, use -1 as index.

How do you check if an element is the last element in a list Python?

The best way to find the last element of the list in Python is using the list[-1]. The list[-n] syntax gets the nth-to-last element. So list[-1] gets the last element, list[-2] gets the second to last. The list[-1] is the most preferable, shortest, and Pythonic way to get the last element.

Are for loops Pythonic?

💡 Python's “for” loops are “for-each” loops I would consider this solution to be quite Pythonic. It's nice and clean and almost reads like pseudo code from a text book. I don't have to keep track of the container's size or a running index to access elements.


1 Answers

Most of the times it is easier (and cheaper) to make the first iteration the special case instead of the last one:

first = True for data in data_list:     if first:         first = False     else:         between_items()      item() 

This will work for any iterable, even for those that have no len():

file = open('/path/to/file') for line in file:     process_line(line)      # No way of telling if this is the last line! 

Apart from that, I don't think there is a generally superior solution as it depends on what you are trying to do. For example, if you are building a string from a list, it's naturally better to use str.join() than using a for loop “with special case”.


Using the same principle but more compact:

for i, line in enumerate(data_list):     if i > 0:         between_items()     item() 

Looks familiar, doesn't it? :)


For @ofko, and others who really need to find out if the current value of an iterable without len() is the last one, you will need to look ahead:

def lookahead(iterable):     """Pass through all values from the given iterable, augmented by the     information if there are more values to come after the current one     (True), or if it is the last value (False).     """     # Get an iterator and pull the first value.     it = iter(iterable)     last = next(it)     # Run the iterator to exhaustion (starting from the second value).     for val in it:         # Report the *previous* value (more to come).         yield last, True         last = val     # Report the last value.     yield last, False 

Then you can use it like this:

>>> for i, has_more in lookahead(range(3)): ...     print(i, has_more) 0 True 1 True 2 False 
like image 95
Ferdinand Beyer Avatar answered Oct 12 '22 12:10

Ferdinand Beyer