开发者

python for loop, how to find next value(object)?

HI, I'm trying to use for loop to find the difference between every two object 开发者_如何学JAVAby minus each other. So, how can I find the next value in a for loop?

for entry in entries:
    first = entry      # Present value
    last = ??????      # The last value how to say?
    diff = last = first


It should be noted that none of these solutions work for generators. For that see Glenn Maynards superior solution.

use zip for small lists:

 for current, last in zip(entries[1:], entries):
     diff = current - last

This makes a copy of the list (and a list of tuples from both copies of the list) so it's good to use itertools for handling larger lists

import itertools as it

items = it.izip(it.islice(entries, 1, None), entries)
for current, last in items:
    diff = current - last

This will avoid both making a copy of the list and making a list of tuples.

Another way to do it without making a copy is

entry_iter = iter(entries)
entry_iter.next() # Throw away the first version
for i, entry in enumerate(entry_iter):
    diff = entry - entries[i]

And yet another way is:

for i in xrange(len(entries) - 1):
    diff = entries[i+1] - entries[i]

This creates an iterator that indexes entries and advances it by one. It then uses enumerate to get an indice with the item. The indice starts at 0 and so points to the previous element because we the loop one item in.

Also, as Tyler pointed out in the comment, a loop might be overkill for such a simple problem if you just want to iterate over the differences.

diffs = (current - last for current, last in 
         it.izip(it.islice(entries, 1, None), entries))


zip works for lists, but for the general case:

def pairs(it):
    it = iter(it)
    prev = next(it)
    for v in it:
        yield prev, v
        prev = v

a = [1,2,3,4,5]
for prev, cur in pairs(a):
    print cur - prev

import itertools as it
for prev, cur in pairs(it.cycle([1,2,3,4])):
    print cur - prev

This works efficiently for large containers, and more importantly, it works for iterators:

for prev, cur in pairs(open("/usr/share/dict/words").xreadlines()):
    print cur, prev,

Edit: I changed the generator to omit the first value with no previous value, since that fits the original question better ("finding differences between adjacent pairs"), and I added an example case showing that it works for an infinitely-repeating iterator.


i don't know exactly what are you looking but maybe this can help :

first = entries[0]
for entry in entries[1:]:
    last = entry       
    diff = last - first 
    first = entry


very simply, using enumerate, no fancy stuff

>>> entries=[10,20,30,40]
>>> for n,i in enumerate(entries):
...     try:
...        print entries[n+1] - i
...     except IndexError:
...        print entries[-1]
...
10
10
10
40
0

上一篇:

下一篇:

精彩评论

暂无评论...
验证码 换一张
取 消

最新问答

问答排行榜