Printing list of list (containing integers) in lexicographic order

Question

Task:You are given three integers x,y and z along with an integer n. You have to print a list of all possible coordinates where the sum of is not equal to n. Print Print the list in lexicographic increasing order.
Below is my code. Works fine except for printing in lexicographic order. Below is my code.Is there a better approach to get lexicographic order of list containing integers?

from itertools import combinations
lst=[]
if __name__ == '__main__':
    x = int(input())
    y = int(input())
    z = int(input())
    n = int(input())
    lst=[[a,b,c]  for b in range(y+1) for c in range(z+1)   for a in range(x+1) ]

finallst=[]
for items in combinations(lst,3):
    for nums in items:
       x=sum(nums)
       if x!=n and nums not in finallst:
            finallst.append(nums)

f_finallst= (sorted(map(str,(finallst)))) #converted to string to get lexicographic order
print (f_finallst) 
My result=['[0, 0, 0]', '[0, 0, 1]', '[0, 1, 0]', '[1, 0, 0]', '[1, 1, 1]']
Expected result=[[0, 0, 0], [0, 0, 1], [0, 1, 0], [1, 0, 0], [1, 1, 1]]

Asked By: Sel_Python

||

Source

Answer 1

print([coords 
       for coords in itertools.product(range(x+1), range(y+1), range(z+1))
       if sum(coords) != n])

Answered By: BallpointBen

Answer 2

By using map(str, finallst), you cast each of the element in the list to a str. You want to keep the elements as they are, but use str as a sorting key.

f_finallst= sorted(finallst, key=str)

Answered By: Olivier Melançon

Answer 3

x = int(input())
y = int(input())
z = int(input())
n = int(input())
lists=[[i,j,k] for i in range(x+1)
               for j in range(y+1)
               for k in range(z+1) if (i+j+k)!=n]
print(lists)

Answered By: Haris Aqeel

Answer 4

I found out that adding the lambda function, which basically works as the key for the sort comparison will do the trick. For a detailed explanation read here

if __name__ == '__main__':
x = int(input())
y = int(input())
z = int(input())
n = int(input())

orginal_list = [[i, j, k] for k in range(z+1) for j in range(y+1) for i in range(x+1) if i + j + k != n]

sorted_list = sorted(orginal_list, key = lambda i: (len(i), i)) 


print(sorted_list)

Answered By: tafadzwabmotsi

Answer 5

x = int(input())
y = int(input())
z = int(input())
n = int(input())
lists=[[i,j,k] for i in range(x+1)
               for j in range(y+1)
               for k in range(z+1) if (i+j+k)!=n]
lists.sort() # to sort it
print(lists)

Answered By: vikiman

Printing list of list (containing integers) in lexicographic order

Question:

Answers: