comparison scripts/process.py @ 1012:01db14e947e4

resolved
author Wendlasida
date Fri, 01 Jun 2018 10:47:49 -0400
parents 933670761a57
children 75af46516b2b
comparison
equal deleted inserted replaced
1011:4f0312bee393 1012:01db14e947e4
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':