FilesExpand file tree

 master

/

ema.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

27 lines (23 loc) · 687 Bytes

 master

/

ema.py

Copy path

Top

File metadata and controls

• Code
• Blame

27 lines (23 loc) · 687 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

from __future__ import print_function

import numpy as np

import matplotlib.pyplot as plt

x = np.arange(5)

print("Exp", np.exp(x))

print("Linspace", np.linspace(-1, 0, 5))

# Calculate weights

N = 5

weights = np.exp(np.linspace(-1., 0., N))

# Normalize weights

weights /= weights.sum()

print("Weights", weights)

c = np.loadtxt('data.csv', delimiter=',', usecols=(6,), unpack=True)

ema = np.convolve(weights, c)[N-1:-N+1]

t = np.arange(N - 1, len(c))

plt.plot(t, c[N-1:], lw=1.0, label='Data')

plt.plot(t, ema, '--', lw=2.0, label='Exponential Moving Average')

plt.title('5 Days Exponential Moving Average')

plt.xlabel('Days')

plt.ylabel('Price ($)')

plt.legend()

plt.grid()

plt.show()