Question

我有一个列表，实际上包含另一个列表的索引。因此，我想从此列表中选择连续的数字

index_list=[3,4,8,9,35,36,37]

我希望将其输出为

[3:4], [8:9], [35:37]

---------------------动机：---------------------

我还有另一个单词主列表，其中有80个单词。

  master_list=['was,'it','to,'go,'I'.........]

因此，连续索引将帮助我从master_list中选择所需的单词

master_list[3:4], master_list[8:9], master_list[35:37]

Answer 1

可能有很多方法可以做到这一点。这是基于reduce()和islice()的一个：

from functools import reduce
from itertools import islice

qw = [3, 4, 8, 9, 12, 13, 14]

master_list = ['Thus', 'the', 'consecutive', 'indices', 'will', 'help', 'me', 'pick', 'out', 'the', 'required', 'words', 'from', 'master_list', 'as']

def divide(value, element):
    if not value[-1] or element - value[-1][-1] == 1:
        value[-1].append(element)
    else:
        value.append([element])

    return value

slices = [(array[0], array[-1]) for array in reduce(divide, qw, [[]])]

print(slices)

for sliced in slices:
    print(list(islice(master_list, *sliced)))

输出

% python3 test.py
[(3, 4), (8, 9), (12, 14)]
['indices']
['out']
['from', 'master_list']
%

请注意，这以惯常的Python方式处理片中的第二个数字，因为它超出了我们想要的数字。如果它确实是您想要的最后一项，则使用+ 1修改此元素：

(array[0], array[-1] + 1)

Answer 2

使用更通用的方法：

i = 0
final_list = []
flag = True
while flag:
    temp_list = [index_list[i]]
    while (i < len(index_list)-1) and (index_list[i+1] - index_list[i]==1):
        i+=1

    temp_list.append(index_list[i])
    final_list.append(temp_list)
    i+=1
    if i >= len(index_list):
        flag=False

print(final_list)

input : [3,4,8,9,35,36,37]
ouptut : [[3, 4], [8, 9], [35, 37]]

input : [1,3,5,7,9]
output : [[1, 1], [3, 3], [5, 5], [7, 7], [9, 9]]

更新：

new_list = []
for i, j in final_list:
    new_list.append(list(range(i,j+1)))
print(new_list)

input:  [3, 4, 5, 16, 17, 31, 32, 33, 34] 
output : [[3, 4, 5], [16, 17], [31, 32, 33, 34]]

如何从列表中选择连续的数字

2 个答案: