二向无常 遇而无往

import pandas as pd
import os

def extract_data_and_save_to_csv(txt_file_path):
    # 加载数据
    try:
        data = pd.read_csv(txt_file_path, skiprows=17, header=None, names=['Potential/V', 'Current/A', 'log(i/A)'])
    except Exception as e:
        print(f"Failed to load data from {txt_file_path}: {str(e)}")
        return

    # 构建输出CSV文件的路径和名称
    output_file_path = os.path.splitext(txt_file_path)[0] + '.csv'

    # 保存到CSV文件
    try:
        data.to_csv(output_file_path, index=False)
        print(f"Data from {txt_file_path} has been successfully saved to {output_file_path}")
    except Exception as e:
        print(f"Failed to save data to {output_file_path}: {str(e)}")

def main(directory_path):
    # 获取目录下所有的TXT文件
    txt_files = [f for f in os.listdir(directory_path) if os.path.isfile(os.path.join(directory_path, f)) and f.lower().endswith('.txt')]

    # 对每个TXT文件执行数据提取和保存操作
    for txt_file in txt_files:
        txt_file_path = os.path.join(directory_path, txt_file)
        extract_data_and_save_to_csv(txt_file_path)

if __name__ == "__main__":
    directory_path = input("Please enter the directory path: ")
    main(directory_path)