python中具有多个循环和if语句的列表理解

Question

I have to print a list of ordered pairs where x+y+z != n. The code doesn't seem to be working

def main():
    x = int(input())
    y = int(input())
    z = int(input())
    n = int(input())
    result = [[i,j,k] for i in range(0,x+1) for j in range(0,y+1) for k in range(0,z+1) if (x+y+z) != n]
    print(result)

Input:

1
1
1
2

Output:

[[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1], [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]]

I shouldn't be getting the ordered pairs [0,1,1] , [1,0,1], [1,1,0].

Please correct me.

Thanks in advance!

Answer 1

You mixed the list comprehension variables with the inputs from the user.

Replace:

if (x+y+z) != n

with

if (i+j+k) != n

Answer 2

One liners are great, but they can confuse me, so I will expand your code

    result = []
    for i in range(0, x+1):
        for j in range(0, y+1):
            for k in range(0, z+1):
                if (i+k+j)!=n:
                    result.append([i,k,j])

And it works for me