Распечатка списка (содержащего целые числа) в лексикографическом порядке

Задача: Вам даны три целых числа x,y и z вместе с целым числом n. Вы должны напечатать список всех возможных координат, где сумма не равна n. Печать Распечатать список в лексикографическом порядке. Ниже мой код. Прекрасно работает, за исключением печати в лексикографическом порядке. Ниже приведен мой код. Есть ли лучший подход для получения лексикографического порядка списка, содержащего целые числа?

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]]

2 ответа

Решение
print([coords 
       for coords in itertools.product(range(x+1), range(y+1), range(z+1))
       if sum(coords) != n])
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)

Используя map(str, finallst), вы приведете каждый элемент в списке к str, Вы хотите сохранить элементы такими, какие они есть, но использовать str в качестве ключа сортировки.

f_finallst= sorted(finallst, key=str)

Я обнаружил, что добавление лямбда-функции, которая в основном работает как ключ для сравнения сортировки, поможет. Подробное объяснение читайте здесь

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)
Другие вопросы по тегам