ZYM/cumcm2024-code
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

write data loader

Browse Source
This commit is contained in:
ZhuangYumin
2024-09-05 21:20:45 +08:00
parent 9528906db8
commit afbb7ade5c
2 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
A/1/A1_to_csv.py
View File

@@ -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])
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="")