File: rec_join_demo.py

package info (click to toggle)
matplotlib 0.98.1-1%2Blenny4
  • links: PTS
  • area: main
  • in suites: lenny
  • size: 18,624 kB
  • ctags: 22,599
  • sloc: python: 76,915; cpp: 63,459; ansic: 5,353; makefile: 111; sh: 12
file content (27 lines) | stat: -rw-r--r-- 664 bytes parent folder | download 
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
 
import numpy as np
import matplotlib.mlab as mlab


r = mlab.csv2rec('data/aapl.csv')
r.sort()
r1 = r[-10:]

# Create a new array 
r2 = np.empty(12, dtype=[('date', '|O4'), ('high', np.float), 
                            ('marker', np.float)])
r2 = r2.view(np.recarray)
r2.date = r.date[-17:-5]
r2.high = r.high[-17:-5]
r2.marker = np.arange(12)

print "r1:"
print mlab.rec2txt(r1)
print "r2:"
print mlab.rec2txt(r2)

defaults = {'marker':-1, 'close':np.NaN, 'low':-4444.}

for s in ('inner', 'outer', 'leftouter'):
    rec = mlab.rec_join(['date', 'high'], r1, r2, 
            jointype=s, defaults=defaults) 
    print "\n%sjoin :\n%s" % (s, mlab.rec2txt(rec))