1.py 955 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. '''
  2. Created on 2019年3月7日
  3. @author: User
  4. '''
  5. from Utils import *
  6. from featureEngine import *
  7. import math
  8. data_file = "temp_data.pk"
  9. data = load(data_file)
  10. print(len(data))
  11. print(data[0])
  12. print(len(data[0][0]))
  13. print(len(data[0][0][0]))
  14. new_data = []
  15. ''''''
  16. for i in range(len(data)):
  17. if len(data[i][0])<=100:
  18. new_data.append(data[i])
  19. print("new data",len(new_data))
  20. for i in range(len(new_data)):
  21. for k in range(len(new_data[i][0])):
  22. for h in range(len(new_data[i][0][k])):
  23. if math.isinf(new_data[i][0][k][h]) or math.isnan(new_data[i][0][k][h]):
  24. new_data[i][0][k][h] = -1
  25. print(new_data[i][0][k][h])
  26. #print(count)
  27. new_data.sort(key=lambda x:x[2])
  28. save(new_data,"sourceData_36Input_"+str(len(new_data))+"_sort.pk")
  29. #new_data = load("sourceData_27Input_25547.pk")
  30. pd_data = paddinig(new_data)
  31. save(pd_data, "data_"+str(len(pd_data[1]))+".pk")