Home > Software engineering >  How to convert list to list of list for adjacent numbers
How to convert list to list of list for adjacent numbers

Time:08-26

  • i have list [31, 32,33, 1,2,3,4, 11,12,13,14]
  • I need to put into adjacent numbers into one list for i, i 1

Expected out [[1,2,3,4], [11,12,13,14], [31, 32,33]]

l = [31, 32,33, 1,2,3,4, 11,12,13,14]
l.sort() #sorted the items
new_l = []
for i in l:
    temp_l = []  # temp list before appending to main list
    if i   1 in l: # if i   1 is present append to temp_list
        temp_l.append(i)
    new_l.append(temp_l) # temp_l has to append to main list

My out is wrong : [[1], [2], [3], [], [11], [12], [13], [], [31], [32], []]

CodePudding user response:

You can use itertools.groupby:

from itertools import groupby

l = [31, 32, 33, 1, 2, 3, 4, 11, 12, 13, 14]
l.sort() 

out = [
    list(v for _, v in g)
    for _, g in groupby(enumerate(l), key=lambda v: v[0] - v[1])
]
print(out)

Prints:

[[1, 2, 3, 4], [11, 12, 13, 14], [31, 32, 33]]

Without itertools.groupby:

l = [31, 32, 33, 1, 2, 3, 4, 11, 12, 13, 14]
l.sort()

out = [[l[0]]]
for i in range(1, len(l)):
    if l[i] - out[-1][-1] == 1:
        out[-1].append(l[i])
    else:
        out.append([l[i]])

print(out)

Edit: Added .sort()

CodePudding user response:

You can append an empty sub-list to the output list when the difference between the current number and the last number in the last sub-list in the output list is not 1, and keep appending the current number to the last sub-list of the output list:

l = [31, 32,33, 1,2,3,4, 11,12,13,14]
l.sort()

output = []
for i in l:
    if not output or i - output[-1][-1] != 1:
        output.append([])
    output[-1].append(i)

output becomes:

[[1, 2, 3, 4], [11, 12, 13, 14], [31, 32, 33]]

Demo: https://replit.com/@blhsing/UnimportantValidTelecommunications

CodePudding user response:

Use a for loop, enumerate() and sort()

l = [31, 32, 33, 1, 2, 3, 4, 11 ,12, 13, 14]
# sort list
l = sorted(l)

sub_l = []
new_list = []
for i, num in enumerate(l):
    # check if not last element
    if not (i   1) == len(l):
        # check if next element is the same as the current   1
        if num   1 == l[i 1]:
            sub_l.append(num)
        else:
            # append sub-list to parent-list and reset sub-list
            new_list.append(sub_l)
            sub_l = []

print(new_list)

Output

[[1, 2, 3], [11, 12, 13], [31, 32]]
  • Related