Python code to generate random "technical analysis" charts

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()