如何按指定表面列表提取数据文件中对应Total CL值（Python正则问题）

问题分析 #

之前的代码仅能匹配首个实例，是因为使用了re.search或re.match这类仅返回单个匹配结果的方法。要提取所有目标表面的Total CL值，需要使用能遍历全部匹配项的正则方法，比如re.finditer或re.findall。

解决方案代码示例 #

假设你的数据文件内容格式如下：

Surface: wing
Total CL: 0.85
Surface: tail
Total CL: 0.12
Surface: body
Total CL: 0.03

以下是可提取指定表面对应Total CL值的代码：

import re

# 读取数据文件内容
with open('aerodynamic_data.txt', 'r') as file:
    data_content = file.read()

# 指定需要提取的目标表面
target_surfaces = {"wing", "body"}

# 预编译正则表达式：匹配表面名称及对应的Total CL值，支持跨行匹配
# 使用命名分组让代码更易读
cl_pattern = re.compile(
    r'Surface:\s*(?P<surface>\w+)\s*Total CL:\s*(?P<cl_value>[\d.]+)',
    re.DOTALL  # 让.匹配换行符，实现跨行匹配
)

# 遍历所有匹配结果，筛选目标表面的数据
surface_cl_data = {}
for match in cl_pattern.finditer(data_content):
    surface_name = match.group("surface")
    if surface_name in target_surfaces:
        surface_cl_data[surface_name] = float(match.group("cl_value"))

# 输出整理后的结果
print("提取到的Total CL值：")
for surface, cl in surface_cl_data.items():
    print(f"{surface}: {cl}")

关键说明 #

• 正则模式：re.DOTALL参数允许正则中的.匹配换行符，确保能匹配到表面名称行和下一行的Total CL值。
• 遍历匹配：re.finditer返回所有匹配结果的迭代器，通过循环可以逐个处理每个表面的数据。
• 筛选目标：通过判断捕获的表面名称是否在指定列表中，只保留需要的数据。

内容的提问来源于stack exchange，提问作者kaush