Question

我知道你可以使用c = cycle(['a', 'b', 'c'])使用c.next()在元素之间循环，但是要获取迭代器的当前元素吗？

例如，如果c.next()返回'c'，则表示迭代器之前处于'b'。我有没有办法'b'而不使用next()？

Answer 1

迭代器/生成器无法获取当前值。你应该保留对它的引用，或者为你创建一个包含它的包装器。

Answer 2

注意：这仅在没有重复的元素的情况下有效。 这是否是一个严重的实际限制取决于每个人的使用。就我而言，与我合作的大部分itertools.cycle都属于此类。

一个人实际上可以使用辅助函数以及其他信息来获取cycle的当前状态。它实际上使用了next，但这对调用方是透明的。

import itertools 

def get_cycle_props(cycle) :
    """Get the properties (elements, length, current state) of a cycle, without advancing it"""
    # Get the current state
    partial = []
    n = 0
    g = next(cycle)
    while ( g not in partial ) :
        partial.append(g)
        g = next(cycle)
        n += 1
    # Cycle until the "current" (now previous) state
    for i in range(n-1) :
        g = next(cycle)
    return (partial, n, partial[0])

def get_cycle_list(cycle) :
    """Get the elements of a cycle, without advancing it"""
    return get_cycle_props(cycle)[0]

def get_cycle_state(cycle) :
    """Get the current state of a cycle, without advancing it"""
    return get_cycle_props(cycle)[2]

def get_cycle_len(cycle) :
    """Get the length of a cycle, without advancing it"""
    return get_cycle_props(cycle)[1]

# initialize list 
test_list = [3, 4, 5, 7, 1] 
c = itertools.cycle(test_list)
print('cycle state =', get_cycle_state(c))
print('cycle length =', get_cycle_len(c))
print('cycle list =', get_cycle_list(c))
next(c)
print('cycle state =', get_cycle_state(c))
print('cycle length =', get_cycle_len(c))
print('cycle list =', get_cycle_list(c))

产生以下输出

cycle state = 3
cycle length = 5
cycle list = [3, 4, 5, 7, 1]
cycle state = 4
cycle length = 5
cycle list = [4, 5, 7, 1, 3]

实际上可以利用此功能通过功能“倒带”一个循环

def shift_cycle(cycle, npos=0) :
    """Shift a cycle, a given number of positions (can be negative)."""
    (cycle_list, nelem, curr_state) = get_cycle_props(cycle)
    for i in range(nelem+npos) :
        g = next(cycle)
    return

尝试

shift_cycle(c, -2)
print('cycle state =', get_cycle_state(c))

让python的itertools循环当前元素

2 个答案: