Mercurial Hosting > traffic-intelligence
comparison trafficintelligence/utils.py @ 1105:e62c2f5e25e6
added sampling function
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Thu, 14 Mar 2019 17:08:05 -0400 |
parents | 7594802f281a |
children | 91faf679e898 |
comparison
equal
deleted
inserted
replaced
1104:1c59091853e0 | 1105:e62c2f5e25e6 |
---|---|
628 headers.append('{}{}'.format(method,c)) | 628 headers.append('{}{}'.format(method,c)) |
629 else: | 629 else: |
630 aggFunctions[method] = aggregationFunction(method) | 630 aggFunctions[method] = aggregationFunction(method) |
631 headers.append(method) | 631 headers.append(method) |
632 return aggFunctions, headers | 632 return aggFunctions, headers |
633 | |
634 def maxSumSample(d, maxSum): | |
635 '''Generates a sample from distribution d (type scipy.stats, using rvs method) | |
636 until the sum of all elements is larger than maxSum''' | |
637 s = 0 # sum | |
638 sample = [] | |
639 while s < maxSum: | |
640 x = d.rvs() | |
641 sample.append(x) | |
642 s += x | |
643 return sample | |
633 | 644 |
634 ######################### | 645 ######################### |
635 # regression analysis using statsmodels (and pandas) | 646 # regression analysis using statsmodels (and pandas) |
636 ######################### | 647 ######################### |
637 | 648 |