Python deep zip

Question

나는 zip과 같은 함수를 작성하려고합니다. 나는 내가 의미하는 바를 설명하는 것에 능숙하지 않다. 그래서 나는 내가하려고하는 것에 대한 '코드'만을 보여줄 것이다.

a = [1,2,3,[4,5]] 
b = a[:] 
zip(a, b) == [(1,1), (2,2), (3,3), ([4,5],[4,5])] 
myzip(a, b) == [(1,1), (2,2), (3,3), [(4,4), (5,5)]] 

나는 너무 익숙하지 않아 재미 없다. 재귀 람다 (recursive lambdas)를 사용하여 간단하게 기능적인 방식으로 작성하여 코드를 더 예쁘게 만들려고합니다. 난 내가 지퍼 이와 비슷한 뭔가를 시도하고있다

def tree_map(func, tree): 
    return map(lambda x: func(x) if not isinstance(x, list) else tree_map(func, x), 
       tree) 

나무에 함수를 매핑 내가 쓴 다른 기능과 출력을 사용하기를 원하기 때문에이 같은 myzip 싶지만 보일 수 없다 그 주위에 내 머리를 감쌌다. 누구든지 myzip을 작성할 수있는 방법에 대한 아이디어가 있습니까?

편집 : tree_map을보세요! 그다지 예쁜 것은 아니다! 나는 적어도 그렇게 생각하지만, 나의 모국어는 Scheme : P 이며, 또한 myzip이 필요로하는만큼 깊게 가고 싶습니다. 기본적으로, 나는 myzip이 나무의 구조를 유지하기를 원합니다. 또한 myzip은 같은 모양의 나무 만 처리합니다.

Answer 1

나는 다음과 같은 일을해야 생각 :

import collections 

def myzip(*args): 
    if all(isinstance(arg, collections.Iterable) for arg in args): 
     return [myzip(*vals) for vals in zip(*args)] 
    return args 

결과 : 나는 유형 검사에 collections.Iterable 대신 list를 사용

>>> a = [1,2,3,[4,[5,6]]] 
>>> b = [1,2,3,[4,[5,6]]] 
>>> myzip(a, b) 
[(1, 1), (2, 2), (3, 3), [(4, 4), [(5, 5), (6, 6)]]] 

주 그렇게 행동이라고 튜플 더 같은 zip() 및 다른 iterables.