luke/autotrader
1
0
Fork 0
mirror of https://github.com/lukewilson2002/autotrader.git synced 2025-08-02 21:19:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add indexed frame shifting

Browse Source
This commit is contained in:
Luke I. Wilson
2023-05-21 12:52:25 -05:00
parent ee67d5c170
commit 30b6482fbf
4 changed files with 228 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
series_indexed.go
View File

@@ -259,6 +259,24 @@ func (s *IndexedSeries[I]) SetValueIndex(index I, val any) *IndexedSeries[I] {
return s.SetValue(row, val)
}
func (s *IndexedSeries[I]) Shift(periods int, nilValue any) *IndexedSeries[I] {
_ = s.series.Shift(periods, nilValue)
return s
}
func (s *IndexedSeries[I]) ShiftIndex(periods int, step func(prev I, amt int) I) *IndexedSeries[I] {
if periods == 0 {
return s
}
// Shift the indexes.
newIndexes := make(map[I]int, len(s.index))
for index, i := range s.index {
newIndexes[step(index, periods)] = i
}
s.index = newIndexes
return s
}
// Sub subtracts the other series values from this series values. The other series must have the same index type. The values are subtracted by comparing their indexes. For example, subtracting two IndexedSeries that share no indexes will result in no change of values.
func (s *IndexedSeries[I]) Sub(other *IndexedSeries[I]) *IndexedSeries[I] {
for index, row := range s.index {