如何在保持列表层次结构的同时连接嵌套列表中的字符串？

Question

我有这张单子：

["a", "b", ["c", ["d", "e"], "f"]]

我希望将每个列表中的字符串元素连接在一起，同时保持列表结构。

预期的行为如下所示：

>>> test_list = ["a", "b", ["c", ["d", "e"], "f"]]
>>> function(test_list)
["ab", ["c", ["de"], "f"]]

它将与此非常相似，但它必须能够处理任意数量的嵌套列表。

我试图用一个递归函数来解决它，但没有得到预期的结果。

有什么提示吗？

Answer 1

使用递归，您可以通过以下方式实现

l = ["a", "b", ["c", ["d", "e"], "f"]]


def func(s):
    res = []
    tmp = []
    val = None
    for i in s:
        if type(i)==str:
            tmp.append(i)
        else:
            res.append(''.join(tmp))
            tmp = []
            val = nested2str(i)
            res.append(val)
    if tmp!=[]:
        res.append(''.join(tmp))
    return res

result = func(l)
print(result)

输出

['ab', ['c', ['de'], 'f']]

Answer 2

可以将itertools.groupby与递归一起使用：

from itertools import groupby as gb
d = ["a", "b", ["c", ["d", "e"], "f"]]
def combine(d1):
  return [i for a, b in gb(d1, key=lambda x:isinstance(x, str)) 
          for i in ([''.join(b)] if a else [combine(k) for k in b])]

print(combine(d))

输出：

['ab', ['c', ['de'], 'f']]