From afbb7ade5c032cd5f63ce9ec3795ef1e61d209d9 Mon Sep 17 00:00:00 2001
From: ZhuangYumin <zhuangyumin@sjtu.edu.cn>
Date: Thu, 5 Sep 2024 21:20:45 +0800
Subject: [PATCH] write data loader

---
 .gitignore       |  3 ++-
 A/1/A1_to_csv.py | 18 ++++++++++++++++--
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index c1f7946..c119264 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 *.swp
 *.json
-*.bak
\ No newline at end of file
+*.bak
+*.dat
\ No newline at end of file
diff --git a/A/1/A1_to_csv.py b/A/1/A1_to_csv.py
index 7267c5a..496de55 100644
--- a/A/1/A1_to_csv.py
+++ b/A/1/A1_to_csv.py
@@ -1,5 +1,19 @@
 import json
 with open("A1_res.json", "r") as file:
   content=json.load(file)
-for lst in content:
-  print(lst[0])
\ No newline at end of file
+fout1=open("tmp1.dat","w")
+for node_point in range(225):
+  for time_point in range(301):
+    v=content[time_point][node_point]["v"]
+    print(v,'\t',file=fout1,sep="",end="")
+  print('\n',file=fout1,end="",sep="")
+fout2=open("tmp2.dat","w")
+for node_point in range(225):
+  for time_point in range(301):
+    x=content[time_point][node_point]["node"][0]
+    print(x,'\t',file=fout2,sep="",end="")
+  print('\n',file=fout2,end="",sep="")
+  for time_point in range(301):
+    y=content[time_point][node_point]["node"][1]
+    print(y,'\t',file=fout2,sep="",end="")
+  print('\n',file=fout2,end="",sep="")
\ No newline at end of file