import xlrd
import numpy as np
import pandas as pd
import time
start = time.clock()
train_data = []
for mood in range(4):
print('mood: %s' %mood)
for people in range(12):
print('people: %s' %people)
for data in range(10):
#读取excel文件
data = xlrd.open_workbook('E:train_data/mood_%s/data_%s_%s_%s.xlsx' %(mood+1, people+1, mood+1, data+1))
#获取行
nrows = data.sheets()[0].nrows
#获取列
ncols = data.sheets()[0].ncols
#截取部分数据
a = np.zeros(shape=(32,12000))
for i in range(nrows):
for j in range(ncols):
data_a = data.sheets()[0].cell_value(i,j)
a[i,j] = data_a
#print(b)
#获取到的数据分段
for tupian in range(12):
train_data_a = a[:,(tupian * 1000):((tupian+1) * 1000)]
train_data.append(train_data_a)
pl = np.array(train_data)
train_data_len = len(train_data)
new_train_data = pl.reshape((train_data_len, 32, 1000, 1))
print(new_train_data.shape)
end = time.clock()
print('use_time: %s' %(end-start))
#数据保存成npy格式
np.save('new_train_data',new_train_data)
#读取保存的npy数据
new_train_data_b = np.load('new_train_data.npy')