从匹配索引开始循环遍历列表多次

Question

我需要在 python 中执行以下操作。我有一个字符串列表list，一个要搜索的字符串text，以及包含要打印的元素数量的变量x。我想遍历x 没有。 list 的连续元素，必要时环绕到 list 的前面。

首先，我需要在list 中找到第一个以text 作为子字符串的元素。然后，我将从list 的第一个元素开始在 匹配的元素之后，并继续遍历list 的总共x 连续元素，必要时环绕。 p>

我该怎么做？

x = 11
text = "string5"
list = ["string1", "string2", "string3", "string4", "string5", "string6", "string7"]

# not sure what to do here...
for elem in list:
    if text in elem:
        #iterate through list, begin with elem and get the next 11 elements
        #once you've reached string7, start over with string1`

在本例中，我希望最终查看以下 11 个元素：

string6
string7
string1
string2
string3
string4
string5
string6
string7
string1
string2

Answer 1

您可以使用cycle from itertools，也可以与islice 和enumerate 结合使用。

from itertools import cycle, islice

x = 11
text = "string5"
lst = ["string1", "string2", "string3", "string4", "string5", "string6", "string7"]
for i, elem in enumerate(lst):
    if text in elem:
        next11 = list(islice(cycle(lst), i+1, i+1+x))
        print(next11)
        print(len(next11))

输出：

['string6', 'string7', 'string1', 'string2', 'string3', 'string4', 'string5', 'string6', 'string7', 'string1', 'string2']
11

Answer 2

对于我们的问题数据：

x = 11 #no. of elements to get
text = 'string5' #text to search in elements of list
lst = ['string1', 'string2', 'string3', 'string4', 'string5', 'string6', 'string7'] 
n = len(lst) 

# index of required text
i = lst.index(text)

算法 1：（循环和切片）

大多数pythonic方法是使用itertools模块offcourse中的islice和cycle函数：

• 无限循环/重复列表（cycle 函数）
• 使用重复列表中的开始/停止索引切出/选择所需元素（slice 函数）

代码：

from itertools import cycle, islice 
desired = list( islice( cycle( lst), i+1, i+1+x))

算法 2：（循环和模索引）

更传统的方式是：

• 遍历列表x 没有。在匹配索引之后开始的时间，i+1
• 使用modulo 索引循环遍历列表元素

代码：

desired = [ lst[ j%n] for j in range(i+1, i+1+x)] 

算法 3 - 错误的实现：（循环和重复）

许多糟糕/缓慢的实现也是可能的，例如使用numpy 函数，如roll、tile：

• 滚动/旋转numpy 数组，使其从匹配索引处的元素开始，i
• 平铺/重复数组以增加其长度超过所需的x 编号。元素数
• 选择x 没有。元素数

代码：

def nextx(lst,i,n,x):
    ll = np.array(lst)
    rll = np.roll(ll, n-i)
    trll = np.tile(rll, x%n+1)
    return list(trll[:x])

输出：

>>> nextx(lst,5,7,11)
['string6', 'string7', 'string1', 'string2', 'string3', 'string4', 'string5', 'string6', 'string7', 'string1', 'string2']
>>> nextx(lst,5,7,11) == [lst[j%len(lst)] for j in range(5,5+11)] == list(islice(cycle(lst),5,5+11))
True

时间：

iPython 线魔术函数%timeit 显示Algorithm 1 与~1.5 and ~11 times faster 分别比2 和3 差一点：

>>> %timeit list( islice( cycle( lst), 5, 5+11))
100000 loops, best of 3: 1.83 µs per loop

>>> %timeit [ lst[ j%len(lst)] for j in range(5, 5+11)] 
100000 loops, best of 3: 2.76 µs per loop

>>> %timeit nextx(lst,5,7,11)
10000 loops, best of 3: 20.6 µs per loop

结论：

只要您可以使用cycle、islice、count、repeat 等，使用itertools 是最快、最流畅的方式！

Answer 3

希望这会有所帮助

>>> x = 11
>>> text = "string5"
>>> list = ["string1", "string2", "string3", "string4", "string5", "string6", "string7"]

>>> for i in range(len(list)):
        if text in list[i]:
            a=i+1;
            break;

>>> L = [list[i%len(list)] for i in range(a, a+x)]

>>> L
['string6', 'string7', 'string1', 'string2', 'string3', 'string4', 'string5', 'string6', 'string7',     'string1', 'string2']

Answer 4

annoid 没有得到它，所以这是一个漂亮的解决方案.. ：P

>>> x = 11
>>> text = "string5"
>>> list = ["string1", "string2", "string3", "string4", "string5", "string6", "string7"]
>>> offset = list.index(text)
>>> list_len = len(list)
>>> for i in range(x):
>>>     print(list[(i+offset)%list_len])

它假定列表中的每个字符串都是唯一的，否则你可以设置偏移量。

Answer 5

使用while循环？

i = 0
n = 0
while 1:
  if i > len(list):
    i = 0
  if (k > x):
    break
  print(list[i])
  i += 1
  k += 1