Question

Possible Duplicate:
Remove leading and trailing zeros from multidimensional list in Python

if I have a list such as:

my_list = [[1,2,0,1], [1,0,0,1]]

I want to split this at the zeros and throw them away, so that I end up with something like:

my_list = [[[1, 2], [1]], [[1],[1]]]

Any help much appreciated.

Answer 1

Divide an conquer, you can use a list comprehension to transform it to a simpler problem

def split_a_list_at_zeros(L):
    ...
    return the_split_list

my_list = [split_a_list_at_zeros(item) for item in my_list]

and Akavall gives a great hint to split the list

from itertools import groupby
def split_a_list_at_zeros(L):
    return [list(g) for k, g in groupby(L, key=lambda x:x!=0) if k]