p_data_process.py 670 B

123456789101112131415161718192021
  1. """
  2. 生成预测数据, 3月14日 和 3月17日的小时级数据
  3. 3月18日至 3月21日的daily 数据
  4. """
  5. import os
  6. import json
  7. from functions import generate_hourly_strings
  8. target_string_list = generate_hourly_strings(start_date="2024031400", end_date="2024031723")
  9. path = "temp_data/data"
  10. L = []
  11. for file in target_string_list:
  12. json_path = os.path.join(path, file + ".json")
  13. print(json_path)
  14. with open(json_path, encoding="utf-8") as f:
  15. data = json.loads(f.read())
  16. for obj in data:
  17. L.append(obj)
  18. with open("prid_data/train_0314_0317.json", "w", encoding="utf-8") as f:
  19. f.write(json.dumps(L, ensure_ascii=False))