将文件夹及子文件夹中二进制bin文件每8个字节转浮点后,按1、2、3的顺序分别放到csv的1、2、3列后转存到csv,csv放到命名为csv文件夹的目录下,并以bin文件的名字+原始文件这几个字命名,然后在将1、2、3列分别进行特征计算,包括均值、方差、rms、小波1~16,记录在另一个csv文件并以bin文件的名字+特征文件这几个字命名。
时间: 2024-04-13 16:30:41 浏览: 110
你可以使用 Python 的标准库和一些第三方库来完成这个任务。下面是一个示例代码,可以实现你的需求:
```python
import os
import csv
import struct
import numpy as np
import pywt
# 创建保存CSV文件的文件夹
csv_folder = "csv"
if not os.path.exists(csv_folder):
os.makedirs(csv_folder)
# 遍历文件夹及子文件夹中的二进制bin文件
for root, dirs, files in os.walk(".", topdown=True):
for filename in files:
if filename.endswith(".bin"):
# 构建CSV文件路径
bin_filepath = os.path.join(root, filename)
csv_filename = f"{os.path.splitext(filename)[0]}_原始文件.csv"
csv_filepath = os.path.join(csv_folder, csv_filename)
# 打开二进制bin文件并读取数据
with open(bin_filepath, "rb") as bin_file:
bin_data = bin_file.read()
# 解析二进制数据并转换为浮点数
floats = []
for i in range(0, len(bin_data), 8):
float_value = struct.unpack("d", bin_data[i:i+8])[0]
floats.append(float_value)
# 将数据按顺序分别放到1、2、3列并保存到CSV文件
with open(csv_filepath, "w", newline="") as csv_file:
writer = csv.writer(csv_file)
writer.writerow(["1", "2", "3"])
for i in range(0, len(floats), 3):
row = [floats[i], floats[i+1], floats[i+2]]
writer.writerow(row)
# 计算特征并保存到特征文件
feature_filename = f"{os.path.splitext(filename)[0]}_特征文件.csv"
feature_filepath = os.path.join(csv_folder, feature_filename)
with open(feature_filepath, "w", newline="") as feature_file:
writer = csv.writer(feature_file)
writer.writerow(["均值", "方差", "rms"] + [f"小波{i}" for i in range(1, 17)])
# 计算特征
data = np.array(floats)
mean = np.mean(data)
variance = np.var(data)
rms = np.sqrt(np.mean(np.square(data)))
wavelet_coeffs = pywt.wavedec(data, "db4", level=16)
# 将特征写入CSV文件
writer.writerow([mean, variance, rms] + list(wavelet_coeffs))
```
这段代码会遍历当前目录及其子目录中的二进制bin文件,将每8个字节转换为浮点数,并将其按顺序放到CSV文件的1、2、3列。然后,它会计算这些数据的均值、方差、rms和小波变换系数,并将这些特征保存到另一个CSV文件中。最后,所有生成的CSV文件都会保存在一个名为"csv"的文件夹中。
阅读全文