FilesExpand file tree

 master

/

sma.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

20 lines (17 loc) · 497 Bytes

 master

/

sma.py

Copy path

Top

File metadata and controls

• Code
• Blame

20 lines (17 loc) · 497 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

from __future__ import print_function

import numpy as np

import matplotlib.pyplot as plt

N = 5

weights = np.ones(N) / N

print("Weights", weights)

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

sma = 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, sma, '--', lw=2.0, label="Moving average")

plt.title("5 Day Moving Average")

plt.xlabel("Days")

plt.ylabel("Price ($)")

plt.grid()

plt.legend()

plt.show()