Return sublists of List of lists

Question

Well, I have this example:

mylist = [[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], [0, 1, 2, 3, 4,          


        

Answer 1

A functional approach:

map(operator.itemgetter(slice(0, 5)), mylist)

Answer 2

Well, you could do the same using list comprehension so at least it would get a bit shorter.

return [x[:5] for x in mylist]

Or if making your function a generator instead, you could also yield the individual elements:

for x in mylist:
    yield x[:5]

Answer 3

You should probably use generators for the inner lists as well, if you are trying to get the process to scale :

g = lambda k: (k[i] for i in range(5))
for x in mylist:
    yield g(x)

Answer 4

Another way to achieve it:

return map(lambda lst: lst[:5], mylist)