如何在python中从另一个带有“begin”和“end”标志的列表构建嵌套列表？-解网

问：

我正在尝试解析一个文件，该文件具有其包含的数据的嵌套结构。数字嵌套在深度上是任意的，每个嵌套都由字符串分隔，例如“begin”和“end”。我想构造一个列表（或字典等），以保留文件中数据结构的嵌套。

举个简单的例子，从一个看起来像这样的列表：

input = [0, 'begin', 1, 2, 'begin', 3, 4, 'end', 5, 'end', 6]

我想去：

[0, [1, 2, [3, 4], 5], 6]

我尝试创建一个递归函数来执行此操作，但无法获得正确的输出。（我会包括我对解决方案的尝试，但由于递归，我无法判断我是否接近）。这个问题应该如何解决？

编辑以添加我的尝试（我希望这不仅会让事情变得混乱）：

input = [0, 'begin', 1, 2, 'begin', 3, 4, 'end', 5, 'end', 6]

def create_nest(items, nest):
    for i, item in enumerate(items):
        if item == 'begin':
            nest.append(create_nest(items[i+1:], []))
        elif item == 'end':
            return nest
        else:
            nest.append(item)

nested_list = []
create_nest(input, nested_list)

这将输出：

[0, [1, 2, [3, 4], 3, 4], 1, 2, [3, 4], 3, 4]

Python 递归嵌套

def to_nested(inpt, out=None):
    it = iter(inpt)   # use an iterator
    if out is None: # optional: just to avoid using a list as default
        out = []    #
    for item in it:
        if item == 'begin': # if begin, let's start a new level
            out.append([])  # add a new list and pass this recursively
            to_nested(it, out=out[-1])
        elif item == 'end': # if end, let's finish this recursion
            return
        else:               # else, add an item to the current level
            out.append(item)
    return out

to_nested([0, 'begin', 1, 2, 'begin', 3, 4, 'end', 5, 'end', 6])
# [0, [1, 2, [3, 4], 5], 6]

请注意，这假定格式正确（结尾数与开头数一样多）。如果没有，您将不得不跟踪它们。

像我一样使用迭代器修改您的方法

inpt = [0, 'begin', 1, 2, 'begin', 3, 4, 'end', 5, 'end', 6]

def create_nest(items, nest):
    items = iter(items)
    for item in items:
        if item == 'begin':
            nest.append(create_nest(items, []))
        elif item == 'end':
            return nest
        else:
            nest.append(item)
    return nest

nested_list = []
create_nest(inpt, nested_list)
# [0, [1, 2, [3, 4], 5], 6]

def collapse(lst):
    save = []
    curr = []
    for x in lst:
        if x == 'begin':
            save.append(curr) # remember what you had so far
            curr = []         # start over
        elif x == 'end':
            old = save.pop()  # get back what you had
            old.append(curr)  # add list you just finished
            curr = old        # pick up where you left off
        else:
            curr.append(x)
    return curr

def parse(tokens):
  r = [[]]
  for t in tokens:
    match t:
      case 'begin':
        r.append([])
      case 'end':
        if len(r) < 2:
          raise RuntimeError("unexpected 'end'")
        r[-2].append(r.pop())
      case _:
        r[-1].append(t)
  if len(r) > 1:
    raise RuntimeError("expected 'end'")
  return r[0]

parse([0, 'end', 'begin', 1, 2, 'begin', 3, 4, 'end', 5, 'end', 6])
# RuntimeError: unexpected 'end'

parse([0, 'begin', 1, 2, 'begin', 3, 4, 'end', 5, 6])
# RuntimeError: expected 'end'

尽管它们可能很优雅，但这里的递归解决方案为格式错误的输入提供了不正确的输出。

上一个：如何检查两个查询是否成功执行？

下一个：从 Powershell 中的密钥（嵌套字典中）获取值

如何在python中从另一个带有“begin”和“end”标志的列表构建嵌套列表？

How to construct a nested list from another list with "begin" and "end" flags in python?

评论

像我一样使用迭代器修改您的方法

评论

评论