aZhaoT commited on
Commit
17bf45d
·
1 Parent(s): 258887b

add total collection period display

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -14,7 +14,12 @@ def load_data():
14
  for name in dataset_names:
15
  print("Loading dataset: ", name)
16
  all_datasets[name] = pd.DataFrame(datasets.load_dataset("cyberorigin/"+name)['train'])
17
- return all_datasets
 
 
 
 
 
18
 
19
  @st.fragment
20
  def visualize(data):
@@ -137,5 +142,7 @@ def visualize(data):
137
 
138
 
139
  st.title("CyberOrigin Data Visualization")
140
- data = load_data()
 
 
141
  visualize(data)
 
14
  for name in dataset_names:
15
  print("Loading dataset: ", name)
16
  all_datasets[name] = pd.DataFrame(datasets.load_dataset("cyberorigin/"+name)['train'])
17
+ total_period = 0
18
+ for dataset in all_datasets.values():
19
+ # dataset["timestamp"] = dataset["timestamp"].astype(float)
20
+ traj_period = dataset["timestamp"].iloc[-1] - dataset["timestamp"].iloc[0]
21
+ total_period += traj_period
22
+ return all_datasets, total_period
23
 
24
  @st.fragment
25
  def visualize(data):
 
142
 
143
 
144
  st.title("CyberOrigin Data Visualization")
145
+ data, period = load_data()
146
+ # display the total period of the data up to 2 decimal places
147
+ st.write("Total period of data: ", round(period, 2), " seconds")
148
  visualize(data)