1
0
Fork 0
master
Pietro 4 years ago
parent 16f9b2d7c3
commit 55a3bdabb9
  1. 5
      PyCTBN/PyCTBN/structure_graph/trajectory_generator.py

@ -66,10 +66,7 @@ class TrajectoryGenerator(object):
t = time[next] t = time[next]
if (max_tr != -1 and n_tr == max_tr) or (t_end != -1 and t >= t_end): if (max_tr != -1 and n_tr == max_tr) or (t_end != -1 and t >= t_end):
last_row = pd.DataFrame(sigma[-1:].values, columns = sigma.columns) sigma.loc[len(sigma) - 1, self._vnames] = -1
last_row.loc[0].values[:] = -1
last_row.loc[0].at["Time"] = round(t, 4)
sigma = sigma.append(last_row, ignore_index = True)
self._generated_trajectory = sigma self._generated_trajectory = sigma
return sigma return sigma
else: else: