Turning a list into nested lists in python

Question

Possible Duplicate:
How can I turn a list into an array in python?

How can I turn a list such as:

data_list = [0,1,2,3,4,5,6,7,8]

into a list of lists such as:

new_list = [ [0,1,2] , [3,4,5] , [6,7,8] ]

ie I want to group ordered elements in a list and keep them in an ordered list. How can I do this?

Sorry - I can't figure out the criterium for defining the subgroups on yur question. In your example, all numbers are "ordered" - so do you want ordered elements AND subgroups of max-lenght 3? Or just lenght 3 for each subgroup? — jsbueno
– jsbueno, Commented Jul 7, 2011 at 17:56
@Sentinel: Not strict duplicate, because does not involve numpy. But perhaps close enough. Thanks — eat
– eat, Commented Jul 7, 2011 at 17:58
This isn't a duplicate of the referenced question at all. The solutions for list and numpy.array are entirely different. — pillmuncher
– pillmuncher, Commented Jul 7, 2011 at 18:05
@Sentinel: You're correct that it is similar but I needed to use lists and not arrays now. When I used the np.reshape method from the previous question to revert back to a 1D array, I lost all internal grouping -- my data was not simply [1,2,3...] but a lot of 3-tuples. I needed a way to put back the organization into threes not using arrays. I think this question is thus useful on its own merit and is significantly different than the previous one. — Double AA
– Double AA, Commented Jul 7, 2011 at 18:08

jsbueno · Accepted Answer · 2011-07-07 17:57:49Z

90

This groups each 3 elements in the order they appear:

new_list = [data_list[i:i+3] for i in range(0, len(data_list), 3)]

Give us a better example if it is not what you want.

answered Jul 7, 2011 at 17:57

jsbueno

113k11 gold badges159 silver badges239 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Todd Over a year ago

I ran into a similar situation to the above question and your code worked perfect! I've been struggling at the last stage. So glad that I could address this.

Graham · Accepted Answer · 2011-07-07 17:54:23Z

23

This assumes that data_list has a length that is a multiple of three

i=0
new_list=[]
while i<len(data_list):
  new_list.append(data_list[i:i+3])
  i+=3

answered Jul 7, 2011 at 17:54

Graham

1,0841 gold badge8 silver badges12 bronze badges

Comments

Hyperboreus · Accepted Answer · 2011-07-07 17:53:29Z

8

Something like:

map (lambda x: data_list[3*x:(x+1)*3], range (3))

answered Jul 7, 2011 at 17:53

Hyperboreus

32.5k9 gold badges50 silver badges88 bronze badges

Comments

Cœur · Accepted Answer · 2018-07-12 16:42:30Z

5

Based on the answer from Fred Foo, if you're already using numpy, you may use reshape to get a 2d array without copying the data:

import numpy
new_list = numpy.array(data_list).reshape(-1, 3)

answered Jul 12, 2018 at 16:42

Cœur

39k25 gold badges207 silver badges282 bronze badges

1 Comment

Nils Mackay Over a year ago

I like this, but it only works when the array has a number of elements that's a multiple of 3, otherwise it will throw a ValueError.

ba__friend · Accepted Answer · 2011-07-07 17:54:37Z

4

new_list = [data_list[x:x+3] for x in range(0, len(data_list) - 2, 3)]

List comprehensions for the win :)

answered Jul 7, 2011 at 17:54

ba__friend

5,9072 gold badges29 silver badges20 bronze badges

Comments

score 2 · Accepted Answer · 2020-10-11 17:36:12Z

The following function expands the original context to include any desired list of lists structure:

def gen_list_of_lists(original_list, new_structure):
    assert len(original_list) == sum(new_structure), \
    "The number of elements in the original list and desired structure don't match"
        
    list_of_lists = [[original_list[i + sum(new_structure[:j])] for i in range(new_structure[j])] \
                     for j in range(len(new_structure))]
        
    return list_of_lists

Using the above:

data_list = [0,1,2,3,4,5,6,7,8]
    
new_list = gen_list_of_lists(original_list=data_list, new_structure=[3,3,3])
# The original desired outcome of [[0,1,2], [3,4,5], [6,7,8]]

new_list = gen_list_of_lists(original_list=data_list, new_structure=[2,3,3,1])
# [[0, 1], [2, 3, 4], [5, 6, 7], [8]]

Sindhukumari Parameswaran · Accepted Answer · 2021-04-10 05:37:33Z

0

The below one is more optimized and quite straightforward.

data_list = [0,1,2,3,4,5,6,7,8]
result =[]
i=0
while i <(len(data_list)-2):
    result.append(data_list[i:i+3])
    i+=3
print(result)

**output**

[[0, 1, 2], [3, 4, 5], [6, 7, 8]]

edited Apr 10, 2021 at 5:37

answered Oct 19, 2020 at 23:46

Sindhukumari Parameswaran

3503 silver badges6 bronze badges

3 Comments

Joe Ferndz Over a year ago

I fixed the code. lots of errors. However, I am not sure if this solves the problem. downvoting the answer.

Sindhukumari Parameswaran Over a year ago

@JoeFerndz can i know what you have fixed in this ? have you run the code?

Sindhukumari Parameswaran Over a year ago

Because I remember writing the working code. What changes have you done?

Sithembiso McGarethson · Accepted Answer · 2023-01-26 05:58:31Z

0

Here is a generalized solution

import math

data_list = [0,1,2,3,4,5,6,7,8]
batch_size=3

n_batches=math.ceil(len(data_list)/batch_size)

[data_list[x*batch_size:min(x*batch_size+batch_size,len(data_list))] 
    for x in range(n_batches)]

It works even if the last sublist is not the same size as the rest (<batch_size)

answered Jan 26, 2023 at 5:58

Sithembiso McGarethson

1

Comments

Manny D · Accepted Answer · 2011-07-07 17:51:59Z

-1

Do you have any sort of selection criteria from your original list?

Python does allow you to do this:

new_list = []
new_list.append(data_list[:3])
new_list.append(data_list[3:6])
new_list.append(data_list[6:])

print new_list
# Output:  [ [0,1,2] , [3,4,5] , [6,7,8] ]

answered Jul 7, 2011 at 17:51

Manny D

20.9k2 gold badges31 silver badges31 bronze badges

2 Comments

eat Over a year ago

FWIW, this won't be very practical one, when you try to generalize the pattern OP described. Thanks

Manny D Over a year ago

Oh I didn't know if the OP was wondering if embedded lists were possible in Python or if there was a way to convert a list into one.

Collectives™ on Stack Overflow

Turning a list into nested lists in python

9 Answers 9

1 Comment

Comments

Comments

1 Comment

Comments

Comments

3 Comments

Comments

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

9 Answers 9

1 Comment

Comments

Comments

1 Comment

Comments

Comments

3 Comments

Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related