Python：使用正則表達式在大文本中選擇文本

Question

s="""set(Q)
    {
information 1
    }

set(CP)
    {
information 2
    }
set(R)
    {
information 3
    }
"""

如何制作一個函數 f ，它將一個標識符作為輸入，如 (CP , R) 來選擇精確設置

示例：f("R") ------> set(R) { 信息 3 }

注意：當我使用此代碼時：它給了我：set(Q) { information 1 }

設置（CP）{信息2}設置（R）

import re
def f(identifier) :

re.findall(r"set\({}\.+set)".format(identifier),s)

Answer 1

您必須使用 re.finditer 來獲取搜索模式的索引，然后循環遍歷字符串直到} ，同時保存您需要的內容

import re

s = """set(Q)
    {
information 1
    }

set(CP)
    {
information 2
    }
set(R)
    {
information 3
    }
set(Q)
    {
information 13
    }
"""
# print (s)


def f(identifier=''):
    var_to_return = []
    search_result = re.finditer(r'set\('+identifier+r'\)', s)
    for result in search_result:
        str_to_save = ''
        idx = result.span()[1]
        while True:
            str_to_save += s[idx]
            idx += 1
            if str_to_save[-1] == '}':
                break

        var_to_return.append(str_to_save)
    return var_to_return


print(f('Q'))

打印一個包含找到的元素的數組：

['\\n {\\n信息 1\\n }', '\\n {\\n信息 13\\n }']