Преобразование вложенных итераций в список

Question

Есть ли простой способ в python (используя itertools или иначе) для преобразования вложенного итерабельного f в соответствующий список или кортеж? Я хотел бы сохранить f, поэтому я могу перебирать его несколько раз, а это значит, что если некоторые вложенные элементы f являются генераторами, у меня будут проблемы.

Я приведу пример ввода/вывода.

>>> g = iter(range(2)) 
>>> my_input = [1, [2, 3], ((4), 5), [6, g]] 
>>> magical_function(my_input) 
[1, [2, 3], [[4], 5], [6, [0, 1]]] 

Было бы хорошо, если бы результат состоял из кортежей. Проблема в том, что итерация по g «потребляет» ее, поэтому ее нельзя использовать снова.

Answer 1

Похоже, это было бы лучше, если бы проверить, является ли каждый элемент итерабельным, и вызывать рекурсивную функцию над ним, если он является итерируемым. Подобно тому, как быстро разводного вверх, я хотел бы попробовать что-то вроде:

import collections 

g = iter(range(2)) 
my_input = [1, [2, 3], ((4), 5), [6, g]] 

def unfold(iterable): 
    ret = [] 
    for element in iterable: 
     if isinstance(element, collections.Iterable): 
      ret.append(unfold(element)) 
     else: 
      ret.append(element) 
    return ret 

n = unfold(my_input) 
print(n) 
print(n) 

который возвращает

$ python3 so.py 
[1, [2, 3], [4, 5], [6, [0, 1]]] 
[1, [2, 3], [4, 5], [6, [0, 1]]] 

Это не самый красивый способ, и вы можете найти способы улучшить его (он ставит все в вместо сохранения кортежей), но вот общая идея, которую я бы использовал.