We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Prepare
  2. Python
  3. Basic Data Types
  4. List Comprehensions
  5. Discussions

List Comprehensions

  • Balazs1982
     + 0 comments

    from itertools import product

    def get_coordinates(x: int, y: int, z: int, n: int) -> list:

    result = product(range(x+1), range(y+1), range(z+1))
return [list(v) for v in filter(lambda x: sum(x) != n, result)]

    if name == 'main':

    x = int(input())
y = int(input())
z = int(input())
n = int(input())

print(get_coordinates(x, y, z, n))