diff options
Diffstat (limited to 'pokemontools/helpers.py')
-rw-r--r-- | pokemontools/helpers.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/pokemontools/helpers.py b/pokemontools/helpers.py new file mode 100644 index 0000000..e2f62e0 --- /dev/null +++ b/pokemontools/helpers.py @@ -0,0 +1,14 @@ +""" +Generic functions that should be reusable anywhere in pokemontools. +""" + +def index(seq, f): + """return the index of the first item in seq + where f(item) == True.""" + return next((i for i in xrange(len(seq)) if f(seq[i])), None) + +def grouper(some_list, count=2): + """splits a list into sublists + given: [1, 2, 3, 4] + returns: [[1, 2], [3, 4]]""" + return [some_list[i:i+count] for i in range(0, len(some_list), count)] |