我有一堆雨滴谱txt数据,第一行是时间就是2024-05-10 10:21:00这样的格式,第二行是值,第三行是空格,然后第四行又是2024-05-10 10:21:00,第五行是值,第六行是空格,这样循环往复。给我写一个批量提取这些值的python代码,提取出来的值写入CSV。
import os import csv from datetime import datetimedef simple_extract(input_folder, output_csv):"""简化版的雨滴谱数据提取"""with open(output_csv, 'w', newline='', encoding='utf-8') as csvfile:writer = csv.writer(csvfile)# 写入CSV表头writer.writerow(['time', 'N', 'data_name'])for filename in os.listdir(input_folder):if filename.endswith('.txt'):file_path = os.path.join(input_folder, filename)with open(file_path, 'r', encoding='utf-8') as file:lines = file.readlines()for i in range(0, len(lines), 3):if i + 1 < len(lines):time_str = lines[i].strip()value_str = lines[i + 1].strip()if time_str and value_str:writer.writerow([time_str, value_str, filename])# 使用简化版 if __name__ == "__main__":input_folder = "/一个值"output_csv = "/数浓度.csv"simple_extract(input_folder, output_csv)print("数据提取完成!")