Question

我编写了一个接受，工作和返回简单的非嵌套元组的函数。

例如：

myfun((1,2,3,4)):
... -> logic
return (1,2,3,4) -> the numbers can change, but the shape will be the same

由于逻辑仅适用于单维元组，但在概念上对于每个嵌套级别都是相同的。我想知道是否有办法将((1,2,(3,)),(4,))之类的嵌套元组转换为普通(1,2,3,4)，然后将其转换回((1,2,(3,)),(4,))。

基本上我想要的是解压缩一个通用输入元组，使用它，然后将结果打包成给定形状的相同形状。

有没有Pythonic方法来完成这样的任务？

可能解压缩可以通过递归来解决，但是我不确定“重新包装”部分。

Answer 1

拆包并不难：

def unpack(parent):
    for child in parent:
        if type(child) == tuple:
            yield from unpack(child)
        else:
            yield child

例如，可以做到这一点。

重新包装有点棘手。我想出了以下内容，它有效，但不是非常pythonic，我害怕：

def repack(structured, flat):
    output = []
    global flatlist
    flatlist = list(flat)
    for child in structured:
        if type(child) == tuple:
            output.append(repack(child, flatlist))
        else:
            output.append(flatlist.pop(0))

    return tuple(output)

示例用法是：

nested = ((1, 2, (3,)), (4,))
plain = tuple(unpack(nested))
renested = repack(nested, plain)

希望这有帮助！

Answer 2

这适用于重新包装：

x = (1,(2,3),(4,(5,6)))
y = (9,8,7,6,5,4)

def map_shape(x, y, start=0):
    if type(x) == tuple:
        l = []
        for item in x:
            mapped, n_item = map_shape(item, y[start:])
            start += n_item
            l.append(mapped)
        return tuple(l), start
    else:
        return y[start], start+1

map_shape(x,y)[0]

输出：

(9, (8, 7), (6, (5, 4)))

Answer 3

我提交了我的版本。它使用相同的功能来平坦并重建列表。如果flat为None，它会变平，否则会通过产生元组来重建。

import collections


def restructure(original, flat=None):
    for el in original:
        if isinstance(el, collections.Iterable) and not isinstance(el, (str, bytes)):
            if flat:
                yield tuple(restructure(el, flat))
            else:
                yield from restructure(el)
        else:
            yield next(flat) if flat else el


def gen():
    i = 0
    while True:
        yield i
        i += 1


def myfun(iterable):
    flat = tuple(restructure(iterable))
    # your transformation ..
    flat = gen()  # assigning infinite number generator for testing
    return restructure(iterable, flat=iter(flat))


x = (1, (2, 3), (4, (5, 6)))
print(tuple(y for y in myfun(x)))  # (0, (1, 2), (3, (4, 5)))

解包并重新打包元组（Python 2.x）

3 个答案: