列出理解以找到匹配的括號

Question

我正在嘗試提出一個列表解析，以匹配打開和關閉括號。 到目前為止，我有這兩個語句分別獲取打開和關閉parens的兩個列表

my_str = "hanz(and(franz/bob())+ 7) + tom(2)"

抓取開放式IDX：

[ i for i,c in enumerate(my_str) if c == '(']

# [4, 8, 18, 31]

抓緊封閉IDX：

[ i for i,c in enumerate(my_str) if c == ')']

# [19, 20, 24, 33]

我想要的是一種便捷的理解方法，可以給我一張與每個配對配對配對對應的配對列表。

即

[ ???? for i,c in enumerate(my_str) ???]

# [(4,24), (8,20), (18,19), (31,33)]

Answer 1

就像@Tordek提到的那樣，盡管這不是不可能，但它不是很實際，

為了完整起見，下面是一個解決方案：

my_str = "hanz(and(franz/bob())+ 7) + tom(2)"

 pt_arr = [ 1 if c == '(' else -1 for i,c in enumerate(my_str ) if c == ')' or c == '(']
idx_arr = [ i for i,c in enumerate(my_str ) if c == ')' or c == '(']

[(idx_arr[strt_idx],idx_arr[strt_idx + [j for j,d in enumerate([ sum(pt_arr[strt_idx:i + 1]) for i,c in enumerate(pt_arr) if i >= strt_idx]) if d == 0][0]]) for strt_idx,f in enumerate(pt_arr) if f == 1]

# [(4,24), (8,20), (18,19), (31,33)]

Answer 2

如評論中所述，正確而簡單的方法是使用堆棧：

my_str = "hanz(and(franz/bob())+ 7) + tom(2)"
stack = []
parens = []
for i, c in enumerate(my_str):
    if c == "(":
        stack.append(i)
    elif c == ")":
        parens.append((stack.pop(), i))
print(parens) # [(18, 19), (8, 20), (4, 24), (31, 33)]

但是，如果您對單行代碼的重視程度超過了可讀性或編碼約定，那么您也可以將其添加到具有副作用的列表理解中：

stack = []
parens = [(stack.pop(), i) for i, c in enumerate(my_str)
          if c == "(" and stack.append(i) or c == ")"]
print(parens) # [(18, 19), (8, 20), (4, 24), (31, 33)]

這利用了and和or進行了短路評估的事實，因此僅在c == "("時才append項目，但之后會失敗，因為append返回None ，並且僅在第二個條件c == ")"時才向結果中添加元素c == ")"是真的，啪最近的位置(從堆棧。

至少，這不是對列表理解的完全濫用，因為結果不會被丟棄，而實際上是期望的結果，並且它可能比您擁有的三個列表理解更容易理解（盡管這些工作沒有副作用），但是一個“方便”的方法的更好解決方案是：使它成為一個函數，而不管它有多少行。