Skip to content

Commit c4ad072

Browse files
optimisation on time
1 parent 3d237a2 commit c4ad072

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/py_eddy_tracker/observations/network.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -490,14 +490,16 @@ def event_timeline(self, ax, field=None, method=None, factor=1, colors_mode="rol
490490

491491
# TODO : fill mappables dict
492492
y_seg = dict()
493+
_time = self.time
494+
493495
if field is not None and method != "all":
494496
for i, b0, _ in self.iter_on("segment"):
495497
y = self[field][i]
496498
if y.shape[0] != 0:
497499
y_seg[b0] = y.mean() * factor
498500
mappables = dict()
499501
for i, b0, b1 in self.iter_on("segment"):
500-
x = self.time[i]
502+
x = _time[i]
501503
if x.shape[0] == 0:
502504
continue
503505

@@ -532,7 +534,7 @@ def event_timeline(self, ax, field=None, method=None, factor=1, colors_mode="rol
532534
else y_seg[seg_next]
533535
)
534536
)
535-
ax.plot((x[-1], self.time[i_n]), (y0, y1), **event_kw)[0]
537+
ax.plot((x[-1], _time[i_n]), (y0, y1), **event_kw)[0]
536538
events["merging"].append((x[-1], y0))
537539

538540
if i_p != -1:
@@ -548,7 +550,7 @@ def event_timeline(self, ax, field=None, method=None, factor=1, colors_mode="rol
548550
else y_seg[seg_previous]
549551
)
550552
)
551-
ax.plot((x[0], self.time[i_p]), (y0, y1), **event_kw)[0]
553+
ax.plot((x[0], _time[i_p]), (y0, y1), **event_kw)[0]
552554
events["spliting"].append((x[0], y0))
553555

554556
j += 1

0 commit comments

Comments
 (0)