[Python]list comprehension 내에 for문 반복하여 3차원좌표 생성하기

전체코드

x, y, z, n = (int(input()) for _ in range(4))

print ([[a,b,c] for a in range(0,x+1) for b in range(0,y+1) for c in range(0,z+1) if a + b + c != n ])

설명

x, y, z, n = (int(input()) for _ in range(4))

# 4가지 변수를 for문을 통해 입력받는다. 

# 단순한 반복의 경우 _변수를 사용하는데 이는 _라는 변수가 실질적으로 활용되지 않음을 나타낸다. 

print ([[a,b,c] for a in range(0,x+1) for b in range(0,y+1) for c in range(0,z+1) if a + b + c != n ])

# list comprehension 내에 for문을 반복하여 3차원 좌표 생성한다.

참고: https://www.hackerrank.com/challenges/list-comprehensions/forum