Question

我正在处理python脚本，该脚本需要从for循环中的列表中读取值并对其执行一些操作。当前看起来像这样：

for i in someList:
    # do some operation here

现在，此特定操作需要几个小时才能完成，所以我想做的是使用多线程/子进程并并行处理多个项目。那么有没有办法可以像这样运行for循环：

for i,j,k in someList:
    #do some operation here

我知道可以使用此语法同时遍历多个列表，但是我希望能够使用同一列表的多个项目。我想从循环中得到的是i，j和k的值以以下方式更新：

i = someList[1], j = someList[2], k = someList[3]

在下一次迭代中：

i = someList[4], j = someList[5], k = someList[6]

Answer 1

我们可以编写自定义的迭代n_list类，该类可用于跳跃（默认跳跃= 1，类似于普通列表）迭代：

class n_list:
    def __init__(self, lst, n=1):
        self.n = n
        self.lst = lst
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        l = self.lst[self.index:self.index+self.n:]
        if len(l) == 0:
            raise StopIteration
        self.index += self.n
        # 5 * [None] -> [None,None,None,None,None]
        # ([1,2,3] + 5 * [None]) -> [1,2,3,None,None,None,None,None]
        # ([1,2,3] + 5 * [None])[:5] -> [1,2,3,None,None]
        # tuple([1,2,3,None,None]) -> (1,2,3,None,None)
        return tuple((l + self.n * [None])[:self.n])

测试我们的 n_list ，元组大小为3：

test_list = [10, 20, 30, 40, 50, 60, 70, 80]
for a, b, c in n_list(test_list, 3):
    print('a: ' + str(a) + ', b: ' + str(b) + ', c: ' + str(c))

输出：

a: 10, b: 20, c: 30
a: 40, b: 50, c: 60
a: 70, b: 80, c: None

在python中的单个列表上使用多个变量（用于循环）

1 个答案: