def flatten(input_arr, output_arr=None):
if output_arr is None:
output_arr = []
for ele in input_arr:
if isinstance(ele, Iterable):
flatten(ele, output_arr)
else:
output_arr.append(ele)
return output_arr
def test_flatten(self):
nested_list = [2, 1, [3, [4, 5], 6], 7, [8]]
flattened = flatten(nested_list)
self.assertEqual(flattened, [2, 1, 3, 4, 5, 6, 7, 8])
nested_list = [[3, [4, 5], 6], 7, [8]]
flattened = flatten(nested_list)
self.assertEqual(flattened, [3, 4, 5, 6, 7, 8])
nested_list = [[], [8]]
flattened = flatten(nested_list)
self.assertEqual(flattened, [8])
def test_flatten(self):
nested_list = [2, 1, [3, [4, 5], 6], 7, [8]]
flattened = flatten(nested_list)
self.assertEqual(flattened, [2, 1, 3, 4, 5, 6, 7, 8])
nested_list = [[3, [4, 5], 6], 7, [8]]
flattened = flatten(nested_list)
self.assertEqual(flattened, [3, 4, 5, 6, 7, 8])
nested_list = [[], [8]]
flattened = flatten(nested_list)
self.assertEqual(flattened, [8])
def flatten_iter(iterable):
"""
Takes as input multi dimensional iterable and
returns generator which produces one dimensional output.
"""
for element in iterable:
if isinstance(element, Iterable):
yield from flatten_iter(element)
else:
yield element
a = [1, 2, 3, [2, 3, 5, [5, 8, 3]]]
#out=[1, 2, 3, 2, 3, 5, 5, 8, 3]
print(flatten(a))
c = []
for i in flatten_iter(a):
#print(i)
c.append(i)
print(c)
b = str(a)
alist = []
for i in b:
if i.isdigit():
alist.append(int(i))
from algorithms.arrays import flatten, flatten_iter
from algorithms.arrays import flatten
if __name__ == "__main__":
mydict = {"key": "test", "key2": {"key3": "value"}}
mydict2 = {"a": 1, "c": {"a": 2, "b": {"x": 5, "y": 10}}, "d": [1, 2, 3]}
# mydict = flatten(mydict)
mydict2 = flatten(mydict2)
print(mydict2)
@file: 2_Flatten.py
@time: 2019/1/2 14:41
@desc:
'''
from algorithms.arrays import flatten, flatten_iter
from collections import Iterable
def flatten_list(iterator):
for ele in iterator:
if isinstance(ele, Iterable):
# yield from 返回迭代器的值，而yield则返回迭代器本身
yield from flatten_iter(ele)
else:
yield ele
if __name__ == '__main__':
my_list = [[7, 8], [6, 5, 4], [3, 2, 1]]
result_list = flatten(my_list)
iter_list = flatten_list(my_list)
#print(result_list)
try:
for i in range(81):
print(iter_list.__next__())
except StopIteration:
pass
#print("StopIteration")