Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from math import *
- import random
- import numpy as np
- import pylab as pl
- import sys
- tries = int(sys.argv[1])
- increment = 1.0
- values = np.empty(tries)
- values[0] = 0.0
- for i in range(1,tries):
- flip = random.randint(1, 2)
- if flip == 1:
- values[i] = values[i-1] + increment
- else:
- values[i] = values[i-1] - increment
- #Group each 10 bases into a single day:
- day_int = 50
- ndays = tries / day_int
- day_mins = np.empty(ndays)
- day_maxs = np.empty(ndays)
- day_open = values[::day_int]
- day_close = values[day_int-1::day_int]
- for n in range(0,ndays,1):
- tmp = np.array(values[day_int*n:day_int*n+day_int])
- day_mins[n] = np.min(tmp)
- day_maxs[n] = np.max(tmp)
- deltas = day_maxs - day_mins
- ax = pl.axes()
- rects1 = ax.bar(np.arange(ndays), deltas, 0, color = 'r', bottom = day_mins)
- for opening, closing, bar in zip(day_open, day_close, rects1):
- x, w = bar.get_x(), 0.2
- args = {}
- ax.plot((x - w, x), (opening, opening), **args)
- ax.plot((x, x + w), (closing, closing), **args)
- pl.show()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement