我想用嵌套括号将字符串解析为嵌套列表。
示例 - parse_nested("abc(def(gh)ij)klm") -> ["abc", ["def", ["gh"], "ij"], "klm"]
。
我的代码有效,但我想知道是否有更好的方法。
这里是我的代码(我将编写检查以便稍后捕获格式错误的输入):
def match(s, start, stop):
start = c = s.index(start)
level = 0
while c < len(s):
if s[c] == start:
level += 1
elif s[c] == stop:
level -= 1
if level == 0:
return (start+1, c)
c += 1
return None
def parse_nested(s):
if start not in s and stop not in s:
return s
else:
g = match(s, start, stop)
first = s[:g[0]-1]
middle = parse_nested(s[g[0]:g[1]])
last = parse_nested(s[g[1]+1:])
result = [first] if first else []
if type(middle) == str:
result.append([middle])
else:
result.append(middle)
if last:
if type(last) == str:
result.append(last)
else:
result += last
return result
如果没有使用解析库,还有更短/更好的方法吗?