python 列表迭代器_Python中的扁平化嵌套列表迭代器

假设我们有一个嵌套的整数列表；我们必须实现一个迭代器以使其扁平化。每个元素可以是整数，也可以是列表。该列表的元素也可以是整数或其他列表。因此，如果输入类似于[[1，1]，2，[1，1]]，则输出将为[1，1，2，1，1，1]

为了解决这个问题，我们将遵循以下步骤-在初始化部分，它将获取嵌套列表，其工作方式如下：

将res设置为空列表，索引：= 0，调用getVal(nestedList)

该getVal()会采取nestedIntegers，这将作为工作-

为我在nestedIntegers如果i是整数，则将i插入res数组，否则调用getVal(i列表)

该next()方法将返回索引所指向的值，并将索引增加1

hasNext()当旁边有一个元素时，它将返回true，否则返回false

示例(Python)

让我们看下面的实现以更好地理解-class NestedIterator(object):

def __init__(self, nestedList):

self.res = []

self.index = 0

self.getVal(nestedList)

#print(self.res)

def getVal(self,NestedList):

for item in NestedList:

if isinstance(item, int):

self.res.append(item)

else:

self.getVal(item)

def next(self):

self.index+=1

return self.res[self.index-1]

def hasNext(self):

if self.index == len(self.res):

return False

return True

ob = NestedIterator([[1,1],2,[1,1]])

while ob.hasNext():

print(ob.next())

输入项[[1,1],2,[1,1]]

输出结果1

1

2

1

1