self_example/TensorFlow_eaxmple/datadeal_try/test.py

41 lines
1.3 KiB
Python

'''name='r'+''+E:\data\DDS_data\平行齿轮箱齿轮表面磨损故障恒速\DATA\dds测试故障库4.6#0001.dat+''
print(name)'''
'''创建一个获取文件夹下所有后缀名为.dat得文件'''
import os
import numpy as np
def filename_all(name):
L = []
for root, dir, filename in os.walk(name):
for file in filename:
if os.path.splitext(file)[1] == '.dat':
L.append(file)
for file in L:
print(name + file)
# print(L.shape[0])
'''filename_all('E:\data\DDS_data\平行齿轮箱齿轮表面磨损故障恒速\DATA')
k=r'E:\data\DDS_data\平行齿轮箱齿轮表面磨损故障恒速\DATA'''''
# 读取文件
voiceArray = []
filename = 'E:\data\DDS_data\平行齿轮箱齿轮表面磨损故障恒速\DATA\dds测试故障库4.6#0001.dat'
each_data = np.fromfile(filename, dtype=np.float32)
print("each_data:",each_data)
# 将voiceArray转化为np的数组
data = np.array(each_data)
print("data",data)
print("data_shape:",data.shape)
# 创建一个二维空数组
data_true = [[] for i in range(9)]
# 数据按节点个数分成head个分别存入二维空数组的各行
for i in range(9):
data_true[i] = data[i::9]
# 将二维空数组重新整合成np.array的形式
data_true = np.array(data_true)
print(data_true.shape)
print(data_true)