softBV-calulate-first

Screen/process_txt.py

@@ -0,0 +1,95 @@
+# -*- coding: utf-8 -*-
+
+import os
+import re
+import sys
+import csv
+
+
+def extract_data_from_folder(folder_path):
+    """
+    读取文件夹中所有txt文件，提取指定信息，并写入一个CSV文件。
+    （已修正正则表达式）
+
+    :param folder_path: 包含txt文件的文件夹路径。
+    """
+    # 检查文件夹是否存在
+    if not os.path.isdir(folder_path):
+        print(f"错误：文件夹 '{folder_path}' 不存在。")
+        return
+
+    # 定义要提取数据的正则表达式
+    # 1. 对应 "Percolation diameter (A): 1.234"
+    pattern1 = re.compile(r"Percolation diameter \(A\): ([\d\.]+)")
+
+    # 2. 对应 "the minium of d 1.23 #" (已根据您的反馈修正)
+    pattern2 = re.compile(r"the minium of d\s*([\d\.]+)\s*#")
+
+    # 3. 对应 "Maximum node length detected: 5.67 A"
+    pattern3 = re.compile(r"Maximum node length detected: ([\d\.]+) A")
+
+    # 存储所有提取到的数据
+    all_data = []
+
+    # 遍历文件夹中的所有文件，使用 sorted() 确保处理顺序一致
+    for filename in sorted(os.listdir(folder_path)):
+        if filename.endswith(".txt"):
+            txt_path = os.path.join(folder_path, filename)
+
+            try:
+                with open(txt_path, 'r', encoding='utf-8') as file:
+                    content = file.read()
+
+                    # 使用修正后的正则表达式查找数据
+                    match1 = pattern1.search(content)
+                    match2 = pattern2.search(content)
+                    match3 = pattern3.search(content)
+
+                    # 提取匹配到的值，如果未匹配到则为空字符串 ''
+                    val1 = match1.group(1) if match1 else ''
+                    val2 = match2.group(1) if match2 else ''
+                    val3 = match3.group(1) if match3 else ''
+
+                    # 获取文件名（不含.txt后缀）
+                    base_filename = os.path.splitext(filename)[0]
+
+                    # 将这一行的数据添加到总列表中
+                    all_data.append([base_filename, val1, val2, val3])
+
+            except Exception as e:
+                print(f"处理文件 {filename} 时发生错误: {e}")
+
+    # 如果没有找到任何txt文件或数据，则不创建csv
+    if not all_data:
+        print("未在文件夹中找到任何 .txt 文件或未能提取任何数据。")
+        return
+
+    # 根据文件夹名确定CSV文件名
+    folder_name = os.path.basename(os.path.normpath(folder_path))
+    csv_filename = f"{folder_name}.csv"
+
+    # 写入CSV文件
+    try:
+        with open(csv_filename, 'w', newline='', encoding='utf-8') as csvfile:
+            writer = csv.writer(csvfile)
+
+            # 写入表头
+            headers = ['filename', 'Percolation_diameter_A', 'minium_of_d', 'Max_node_length_A']
+            writer.writerow(headers)
+
+            # 写入所有数据
+            writer.writerows(all_data)
+
+        print(f"数据成功写入到文件: {csv_filename}")
+
+    except Exception as e:
+        print(f"写入CSV文件 {csv_filename} 时发生错误: {e}")
+
+
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print("用法: python your_script_name.py <folder_path>")
+        sys.exit(1)
+
+    input_folder = sys.argv[1]
+    extract_data_from_folder(input_folder)