3

我正在尝试编写像 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)]]

我是如此坚持这一点，它甚至不好笑。我正在尝试使用递归 lambda 以简单的函数式方式编写它，以使我的代码更漂亮。我想要这样的 myzip，因为我想将它的输出与我编写的另一个函数一起使用，该函数将函数映射到树

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

我一直在尝试用 zip 做类似的事情，但我似乎无法理解它。有人对我如何编写 myzip 有任何想法吗？

编辑：看看tree_map！是不是很漂亮！至少我是这么认为的，但我的母语是 Scheme :P，而且，我希望 myzip 尽可能深入。基本上，我希望 myzip 保留我通过它的树的结构。此外，myzip 只会处理形状相同的树。

4

1 回答 1

7

我认为以下应该有效：

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

结果：

>>> 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)]]]

请注意，我在类型检查中使用collections.Iterable而不是，list以便行为更像zip()元组和其他可迭代对象。

于 2012-09-27T21:51:59.787 回答