diff options
Diffstat (limited to 'src/utils.py')
| -rw-r--r-- | src/utils.py | 23 | 
1 files changed, 23 insertions, 0 deletions
| diff --git a/src/utils.py b/src/utils.py new file mode 100644 index 0000000..3ce2048 --- /dev/null +++ b/src/utils.py @@ -0,0 +1,23 @@ +#!/usr/bin/env python +# encoding: utf-8 +""" +utils.py + +Created by Jason Sundram, on 2010-04-05. +""" + +def flatten(l): +    """ Converts a list of tuples to a flat list. +        e.g. flatten([(1,2), (3,4)]) => [1,2,3,4] +    """ +    return [item for pair in l for item in pair] + +def tuples(l, n=2): +    """ returns n-tuples from l. +        e.g. tuples(range(4), n=2) -> [(0, 1), (1, 2), (2, 3)] +    """ +    return zip(*[l[i:] for i in range(n)]) + +def rows(m): +    """returns the # of rows in a numpy matrix""" +    return m.shape[0]
\ No newline at end of file | 
