Skip to content

Conversation

@eromoe
Copy link

@eromoe eromoe commented Apr 4, 2023

For example, I have 1 minute scale tick data, and I'd like add 5 minute/ 15 minute feature upon .

I wrote a function

def apply_over(func, arr, stride):
	n = len(arr)
	s = np.empty(n).reshape(-1, stride)

	for i in range(stride):
		s[:,i] = func(arr[i::stride])

	return s.reshape(n,)

This generate 5 minute SMA features for each 1 minute tick .

arr = np.arange(n).astype(float)
sma_5_5 = apply_over(lambda a: talib.SMA(a, 5) , arr, 5)

you can think arr = np.arange(1000).astype(float) as a stock close price at minute level .
It is a 1000 time tick collection.

On every tick , I need calculate sma5 on

  • 1 minute scale (simply talib.SMA(arr, 5) )
  • 5 minute scale (apply_over(lambda a: talib.SMA(a, 5) , arr, 5) )
  • 15 minute scale (apply_over(lambda a: talib.SMA(a, 5) , arr, 15) )

For example:
A series ...,500, 501, 502, 503, 504 , 505,....

At 505,

  • 1 minute scale = SMA([ 501, 502, 503, 504 , 505])
  • 5 minute scale = SMA([ 485, 490, 495, 500 , 505])

Using pandas resmple(freq='5min').first() would make gaps , shrink arr length from 1000 to 200 . You need do it 5 times with each shift [0,1,2,3,4] and apply SMA to make sure every tick have 5 minute feature, , like what I do in apply_over


Because I really like vectorbt, so share this idea here.

@eromoe
Copy link
Author

eromoe commented Apr 4, 2023


import scipy.stats as ss

arr = ss.halfnorm.rvs(10, 5, (100,))

v1 = vbt.mtalib('SMA', 5).run(arr, 5).real
v2 = vbt.talib('SMA').run(arr, 5).real


dfs = [v1, v2]

fig = go.Figure()
# a = a.set_index('ds').reset_index()
for a in dfs:
    fig.add_trace(
        go.Line(
            x=np.arange(len(a)),
            y=a,
        )
    )
fig.show()

image

Copy link

@catbrower catbrower left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

what is mtalib? Apologies but this entire PR feels like a typo. Requesting changes to clarify.

```
"""
import talib
from talib import abstract

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why are we importing talib here? They should already be available

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants