comparison scripts/process.py @ 998:933670761a57

updated code to python 3 (tests pass and scripts run, but non-executed parts of code are probably still not correct)
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Sun, 27 May 2018 23:22:48 -0400
parents dc0be55e2bf5
children 75af46516b2b
comparison
equal deleted inserted replaced
997:4f3387a242a1 998:933670761a57
1 #! /usr/bin/env python 1 #! /usr/bin/env python3
2 2
3 import sys, argparse 3 import sys, argparse
4 from pathlib2 import Path 4 from pathlib2 import Path
5 5
6 import matplotlib 6 import matplotlib
84 medianSpeeds[vs.cameraView.siteIdx].append(speeds[0]) 84 medianSpeeds[vs.cameraView.siteIdx].append(speeds[0])
85 speeds85[vs.cameraView.siteIdx].append(speeds[1]) 85 speeds85[vs.cameraView.siteIdx].append(speeds[1])
86 for speeds, name in zip([medianSpeeds, speeds85], ['Median', '85th Centile']): 86 for speeds, name in zip([medianSpeeds, speeds85], ['Median', '85th Centile']):
87 plt.ioff() 87 plt.ioff()
88 plt.figure() 88 plt.figure()
89 plt.boxplot(speeds.values(), labels = [session.query(Site).get(siteId).name for siteId in speeds]) 89 plt.boxplot(list(speeds.values()), labels = [session.query(Site).get(siteId).name for siteId in speeds])
90 plt.ylabel(name+' Speeds (km/h)') 90 plt.ylabel(name+' Speeds (km/h)')
91 plt.savefig(name.lower()+'-speeds.png', dpi=150) 91 plt.savefig(name.lower()+'-speeds.png', dpi=150)
92 plt.close() 92 plt.close()
93 93
94 if args.analyze == 'interaction': 94 if args.analyze == 'interaction':