From 74a9f14502a51f25f16be62135d28dc7adba6fbb Mon Sep 17 00:00:00 2001 From: Bryan Bishop Date: Sun, 1 Sep 2013 15:33:52 -0500 Subject: move index and grouper into helpers.py --- pokemontools/helpers.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 pokemontools/helpers.py (limited to 'pokemontools/helpers.py') 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)] -- cgit v1.2.3