慕用2447696
此函数应该能够快速平整嵌套的可迭代容器,而无需使用任何递归:import collectionsdef flatten(iterable): iterator = iter(iterable) array, stack = collections.deque(), collections.deque() while True: try: value = next(iterator) except StopIteration: if not stack: return tuple(array) iterator = stack.pop() else: if not isinstance(value, str) \ and isinstance(value, collections.Iterable): stack.append(iterator) iterator = iter(value) else: array.append(value)大约五年后,我对此事的看法发生了变化,使用它可能会更好:def main(): data = [1, 2, [3, 4, [5], []], [6]] print(list(flatten(data)))def flatten(iterable): iterator, sentinel, stack = iter(iterable), object(), [] while True: value = next(iterator, sentinel) if value is sentinel: if not stack: break iterator = stack.pop() elif isinstance(value, str): yield value else: try: new_iterator = iter(value) except TypeError: yield value else: stack.append(iterator) iterator = new_iteratorif __name__ == '__main__': main()