def window_wise(seq, size):
"""Return all windows of `size` elements in `seq`.
>>> _show (window_wise (range (5), 1))
[(0,), (1,), (2,), (3,), (4,)]
>>> _show (window_wise (range (5), 2))
[(0, 1), (1, 2), (2, 3), (3, 4)]
>>> _show (window_wise (range (5), 3))
[(0, 1, 2), (1, 2, 3), (2, 3, 4)]
>>> _show (window_wise (range (5), 4))
[(0, 1, 2, 3), (1, 2, 3, 4)]
>>> _show (window_wise (range (5), 5))
[(0, 1, 2, 3, 4)]
>>> _show (window_wise (range (5), 6))
[]
"""
from _TFL.DL_List import DL_Ring
s = iter(seq)
h = tuple((next(s) for i in range(size)))
if len(h) == size:
w = DL_Ring(h)
yield tuple(w.values())
while True:
w.pop_front()
w.append(next(s))
yield tuple(w.values())
def window_wise(seq, size):
"""Return all windows of `size` elements in `seq`.
>>> _show (window_wise (range (5), 1))
[(0,), (1,), (2,), (3,), (4,)]
>>> _show (window_wise (range (5), 2))
[(0, 1), (1, 2), (2, 3), (3, 4)]
>>> _show (window_wise (range (5), 3))
[(0, 1, 2), (1, 2, 3), (2, 3, 4)]
>>> _show (window_wise (range (5), 4))
[(0, 1, 2, 3), (1, 2, 3, 4)]
>>> _show (window_wise (range (5), 5))
[(0, 1, 2, 3, 4)]
>>> _show (window_wise (range (5), 6))
[]
"""
from _TFL.DL_List import DL_Ring
s = iter(seq)
try:
h = tuple((next(s) for i in range(size)))
except RuntimeError:
pass
else:
if len(h) == size:
w = DL_Ring(h)
yield tuple(w.values())
while True:
w.pop_front()
try:
nxt = next(s)
except StopIteration:
break
else:
w.append(nxt)
yield tuple(w.values())
def window_wise (seq, size) :
"""Return all windows of `size` elements in `seq`.
>>> _show (window_wise (range (5), 1))
[(0,), (1,), (2,), (3,), (4,)]
>>> _show (window_wise (range (5), 2))
[(0, 1), (1, 2), (2, 3), (3, 4)]
>>> _show (window_wise (range (5), 3))
[(0, 1, 2), (1, 2, 3), (2, 3, 4)]
>>> _show (window_wise (range (5), 4))
[(0, 1, 2, 3), (1, 2, 3, 4)]
>>> _show (window_wise (range (5), 5))
[(0, 1, 2, 3, 4)]
>>> _show (window_wise (range (5), 6))
[]
"""
from _TFL.DL_List import DL_Ring
s = iter (seq)
try :
h = tuple ((next (s) for i in range (size)))
except RuntimeError :
pass
else :
if len (h) == size :
w = DL_Ring (h)
yield tuple (w.values ())
while True:
w.pop_front ()
try :
nxt = next (s)
except StopIteration :
break
else :
w.append (nxt)
yield tuple (w.values ())
def window_wise (seq, size) :
"""Return all windows of `size` elements in `seq`.
>>> _show (window_wise (range (5), 1))
[(0,), (1,), (2,), (3,), (4,)]
>>> _show (window_wise (range (5), 2))
[(0, 1), (1, 2), (2, 3), (3, 4)]
>>> _show (window_wise (range (5), 3))
[(0, 1, 2), (1, 2, 3), (2, 3, 4)]
>>> _show (window_wise (range (5), 4))
[(0, 1, 2, 3), (1, 2, 3, 4)]
>>> _show (window_wise (range (5), 5))
[(0, 1, 2, 3, 4)]
>>> _show (window_wise (range (5), 6))
[]
"""
from _TFL.DL_List import DL_Ring
s = iter (seq)
h = tuple ((next (s) for i in range (size)))
if len (h) == size :
w = DL_Ring (h)
yield tuple (w.values ())
while True:
w.pop_front ()
w.append (next (s))
yield tuple (w.values ())
def moving_average(s, n, central=False):
"""Generator for moving average of `n` data points over sequence `s`.
>>> list (moving_average (range (10), 3))
[(2, 1.0), (3, 2.0), (4, 3.0), (5, 4.0), (6, 5.0), (7, 6.0), (8, 7.0), (9, 8.0)]
>>> list (moving_average (range (10), 4))
[(3, 1.5), (4, 2.5), (5, 3.5), (6, 4.5), (7, 5.5), (8, 6.5), (9, 7.5)]
>>> list (moving_average (range (10), 5))
[(4, 2.0), (5, 3.0), (6, 4.0), (7, 5.0), (8, 6.0), (9, 7.0)]
>>> list (moving_average (range (10), 3, True))
[(1, 1.0), (2, 2.0), (3, 3.0), (4, 4.0), (5, 5.0), (6, 6.0), (7, 7.0), (8, 8.0)]
>>> list (moving_average (range (10), 5, True))
[(2, 2.0), (3, 3.0), (4, 4.0), (5, 5.0), (6, 6.0), (7, 7.0)]
>>> list (moving_average (range (10), 7, True))
[(3, 3.0), (4, 4.0), (5, 5.0), (6, 6.0)]
>>> list (moving_average (range (10), 9, True))
[(4, 4.0), (5, 5.0)]
>>> list (moving_average (range (10), 10, True))
[(5, 4.5)]
>>> list (moving_average ((1.28, 1.31, 1.29, 1.28, 1.30, 1.31, 1.27), 3, True))
[(1, 1.2933333333333332), (2, 1.2933333333333332), (3, 1.2899999999999998), (4, 1.2966666666666664), (5, 1.293333333333333)]
>>> list (moving_average ((1.28, 1.31, 1.29, 1.28, 1.30, 1.31, 1.27), 5, True))
[(2, 1.292), (3, 1.298), (4, 1.29)]
>>> list (moving_average ((1.28, 1.31, 1.29, 1.28, 1.30, 1.31, 1.27), 7, True))
[(3, 1.2914285714285714)]
"""
if central:
i = n // 2
else:
i = n - 1
s = iter(s)
w = DL_Ring(next(s) for k in range(n))
m = float(n)
v = sum(w.values()) / m
yield i, v
for x in s:
i += 1
v += (-w.pop_front() / m) + (x / m)
w.append(x)
yield i, v
def moving_average (s, n) :
"""Generator for moving average of `n` data points over sequence `s`.
>>> def show (ma_s, fmt = "%.1f") :
... print ("[" + ", ".join (fmt % v for v in ma_s) + "]")
>>> show (moving_average (range (10), 2))
[0.5, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5, 8.5]
>>> show (moving_average (range (10), 3))
[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0]
>>> show (moving_average (range (10), 4))
[1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5]
>>> show (moving_average (range (10), 5))
[2.0, 3.0, 4.0, 5.0, 6.0, 7.0]
>>> fmt = "%.2f"
>>> show (moving_average ((1.28, 1.31, 1.29, 1.28, 1.30, 1.31, 1.27), 3), fmt)
[1.29, 1.29, 1.29, 1.30, 1.29]
>>> show (moving_average ((1.28, 1.31, 1.29, 1.28, 1.30, 1.31, 1.27), 5), fmt)
[1.29, 1.30, 1.29]
>>> show (moving_average ((1.28, 1.31, 1.29, 1.28, 1.30, 1.31, 1.27), 7), fmt)
[1.29]
"""
if n == 1 :
for x in s :
yield x
else :
m = float (n)
s = iter (s)
w = DL_Ring (next (s) / m for k in range (n))
ma = sum (w.values ())
yield ma
for x in s :
x_m = x / m
ma -= w.pop_front ()
ma += x_m
w.append (x_m)
yield ma
def moving_average (s, n) :
"""Generator for moving average of `n` data points over sequence `s`.
>>> def show (ma_s, fmt = "%.1f") :
... print ("[" + ", ".join (fmt % v for v in ma_s) + "]")
>>> show (moving_average (range (10), 2))
[0.5, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5, 8.5]
>>> show (moving_average (range (10), 3))
[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0]
>>> show (moving_average (range (10), 4))
[1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5]
>>> show (moving_average (range (10), 5))
[2.0, 3.0, 4.0, 5.0, 6.0, 7.0]
>>> fmt = "%.2f"
>>> show (moving_average ((1.28, 1.31, 1.29, 1.28, 1.30, 1.31, 1.27), 3), fmt)
[1.29, 1.29, 1.29, 1.30, 1.29]
>>> show (moving_average ((1.28, 1.31, 1.29, 1.28, 1.30, 1.31, 1.27), 5), fmt)
[1.29, 1.30, 1.29]
>>> show (moving_average ((1.28, 1.31, 1.29, 1.28, 1.30, 1.31, 1.27), 7), fmt)
[1.29]
"""
if n == 1 :
yield from s
else :
m = float (n)
s = iter (s)
w = DL_Ring (next (s) / m for k in range (n))
ma = sum (w.values ())
yield ma
for x in s :
x_m = x / m
ma -= w.pop_front ()
ma += x_m
w.append (x_m)
yield ma
