Skip to content

Commit b3f93ea

Browse files
committed
processing: use np.diff where relevant
1 parent a6b48d2 commit b3f93ea

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/stagpy/processing.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,9 +34,7 @@ def dtime(sdat: StagyyData) -> Tseries:
3434
dt and time arrays.
3535
"""
3636
time = sdat.tseries.time
37-
return Tseries(
38-
time[1:] - time[:-1], time[:-1], Vart("Time increment dt", "dt", "s")
39-
)
37+
return Tseries(np.diff(time), time[:-1], Vart("Time increment dt", "dt", "s"))
4038

4139

4240
def dt_dt(sdat: StagyyData) -> Tseries:
@@ -53,7 +51,7 @@ def dt_dt(sdat: StagyyData) -> Tseries:
5351
series = sdat.tseries["Tmean"]
5452
temp = series.values
5553
time = series.time
56-
dtdt = (temp[1:] - temp[:-1]) / (time[1:] - time[:-1])
54+
dtdt = np.diff(temp) / np.diff(time)
5755
return Tseries(dtdt, time[:-1], Vart("Derivative of temperature", r"dT/dt", "K/s"))
5856

5957

@@ -116,7 +114,7 @@ def delta_r(step: Step) -> Rprof:
116114
"""
117115
edges = step.rprofs.walls
118116
meta = Varr("Cell thickness", "dr", "m")
119-
return Rprof((edges[1:] - edges[:-1]), step.rprofs.centers, meta)
117+
return Rprof(np.diff(edges), step.rprofs.centers, meta)
120118

121119

122120
def _scale_prof(

0 commit comments

Comments
 (0)