马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
import pandas as pd
import os
import csv
from scipy.signal import find_peaks
import numpy as np
# 全局变量,文件读取路径
read_path = r"C:\Users\Administrator\Desktop\text2"
# 全局变量,处理结果文件输出路径
output_path = r"C:\Users\Administrator\Desktop\text2"
# 获取文件路径
def get_file_path():
read_path = r'C:\Users\Administrator\Desktop\text' #读取文件路径
output_path = r'C:\Users\Administrator\Desktop\text1' #写入结果文件路径
return read_path,output_path
# 读取文件名称和内容
def deal_files():
# 获取read_path下的所有文件名称(顺序读取的)
files = os.listdir(read_path)
if not os.path.exists(output_path):
os.mkdir(output_path)
for file_name in files:
# 读取单个文件内容
dfdata = pd.read_csv(read_path+"\"+file_name)
#处理单个文件(调用方法)
finish_dfdata = get_deal_file(dfdata)
# 输出结果到指定路径下
# finish_dfdata.to_csv(output_path + "\" + "处理结果_" + file_name,index=False)
print("文件处理完毕")
# 处理单个文件程序 /针对不同批量处理文件进行修改对文件的处理代码,返回值:finish_dfdata/
def get_deal_file(dfdata): #以后批量处理任何类型数据只需要改这个函数
col_1 = dfdata["count"] #获取所有Y值
data_1 = np.array(col_1) #把所有Y值建立成一个一维数组
y_mat = list() #建立空列表
#循环 给空列表赋值
for i in range(256): #256是除去所有行,本来想动态,但是一直报错,看到所有文件都是256行,就直接写了数字
y_mat.append(data_1[i]) #把值全赋给这个列表
peaks, _ = find_peaks(y_mat, distance=40) #36行-39行不知道是不是多余,你可以试试把y_mat直接换成data_1数组。 这一行是获取峰值,返回x坐标
# peaks2, _ = find_peaks(y_mat, prominence=100) # BEST!
# peaks4, _ = find_peaks(y_mat, threshold=0.4) # Required vertical distance to its direct neighbouring samples, pretty useless
print(peaks) #上面三个都是提取峰值的功能,参数是来调取,减小震荡,你看看哪个更符合你的需求
return peaks
# 主函数
if __name__=="__main__":
# 获取文件输入和输出路径
read_path,output_path = get_file_path()
# 开始处理文件,并输出处理文件结果
deal_files()
大佬们这个代码我输出成功了,但是输出结果建了一个空的文件夹text1 是什么情况呀 我想把结果保存到excel或者txt里,谢谢大佬
你把第二十八行,也是唯一的一行输出数据到文件的代码 to_csv 注释了,代码怎么可能会有输出,把注释打开再试试看:
import pandas as pd
import os
import csv
from scipy.signal import find_peaks
import numpy as np
# 全局变量,文件读取路径
read_path = r"C:\Users\Administrator\Desktop\text2"
# 全局变量,处理结果文件输出路径
output_path = r"C:\Users\Administrator\Desktop\text2"
# 获取文件路径
def get_file_path():
read_path = r'C:\Users\Administrator\Desktop\text' #读取文件路径
output_path = r'C:\Users\Administrator\Desktop\text1' #写入结果文件路径
return read_path,output_path
# 读取文件名称和内容
def deal_files():
# 获取read_path下的所有文件名称(顺序读取的)
files = os.listdir(read_path)
if not os.path.exists(output_path):
os.mkdir(output_path)
for file_name in files:
# 读取单个文件内容
dfdata = pd.read_csv(read_path+""+file_name)
#处理单个文件(调用方法)
finish_dfdata = get_deal_file(dfdata)
# 输出结果到指定路径下
finish_dfdata.to_csv(output_path + "" + "处理结果_" + file_name,index=False)
print("文件处理完毕")
# 处理单个文件程序 /针对不同批量处理文件进行修改对文件的处理代码,返回值:finish_dfdata/
def get_deal_file(dfdata): #以后批量处理任何类型数据只需要改这个函数
col_1 = dfdata["count"] #获取所有Y值
data_1 = np.array(col_1) #把所有Y值建立成一个一维数组
y_mat = list() #建立空列表
#循环 给空列表赋值
for i in range(256): #256是除去所有行,本来想动态,但是一直报错,看到所有文件都是256行,就直接写了数字
y_mat.append(data_1[i]) #把值全赋给这个列表
peaks, _ = find_peaks(y_mat, distance=40) #36行-39行不知道是不是多余,你可以试试把y_mat直接换成data_1数组。 这一行是获取峰值,返回x坐标
# peaks2, _ = find_peaks(y_mat, prominence=100) # BEST!
# peaks4, _ = find_peaks(y_mat, threshold=0.4) # Required vertical distance to its direct neighbouring samples, pretty useless
print(peaks) #上面三个都是提取峰值的功能,参数是来调取,减小震荡,你看看哪个更符合你的需求
return peaks
# 主函数
if __name__=="__main__":
# 获取文件输入和输出路径
read_path,output_path = get_file_path()
# 开始处理文件,并输出处理文件结果
deal_files()
|