Python 函数封装:提取多个 TXT 文件中 Y 值的最大值及其对应 X 值
以定义函数的区块化编程重新编写上述程序内容:
以下是使用函数封装的区块化编程方式重新编写的程序:
import os
import pandas as pd
def process_txt(file_path):
'从txt文件中读取数据'
df = pd.read_csv(file_path, delimiter='\t', skiprows=1, names=['x', 'y'])
'找到最大值的索引'
max_index = df['y'].idxmax()
'提取最大值和对应的x值'
max_y = df.loc[max_index, 'y']
max_x = df.loc[max_index, 'x']
return max_x, max_y
def main(folder_path, output_path):
'读取文件夹中的txt文件'
txt_files = [file for file in os.listdir(folder_path) if file.endswith('.txt')]
'创建空的DataFrame'
data = pd.DataFrame(columns=['File', 'x_max', 'y_max'])
'逐个读取txt文件'
for file in txt_files:
file_path = os.path.join(folder_path, file)
'处理每个txt文件'
max_x, max_y = process_txt(file_path)
'将结果添加到DataFrame中'
data = data.append({'File': file, 'x_max': max_x, 'y_max': max_y}, ignore_index=True)
'将数据保存到Excel表格'
data.to_excel(output_path, index=False)
# 定义文件夹路径和输出文件路径
folder_path = '文件夹路径'
output_path = '输出文件路径'
# 调用主函数
main(folder_path, output_path)
请将代码中的 '文件夹路径' 替换为您要读取txt文件的实际文件夹路径,并将 '输出文件路径' 替换为您希望保存Excel文件的路径。运行程序后,将会在指定路径下生成一个包含每个文件中y的最大值和对应x值,并标明文件名的Excel表格。这种方式将程序划分为不同的函数,提高了可读性和可维护性。
原文地址: http://www.cveoy.top/t/topic/bBQU 著作权归作者所有。请勿转载和采集!