print the terain

"You are given an array like [5, 4, 3, 2, 1, 3, 4, 0, 3, 4]

Part 1:
Print a terrain where each number represents the height of a column at that index.

 

def print_terrain(heights):
    if not heights:
        return

    max_height = max(heights)

    # Iterate from the highest point down to 1
    for level in range(max_height, 0, -1):
        row = ""
        for h in heights:
            if h >= level:
                row += "+"
            else:
                row += " "
        print(row)

    # Print the base layer (level 0)
    print("+" * len(heights) + " <--- base layer")

# The data
data = [5, 4, 3, 2, 1, 3, 4, 0, 3, 4]
print_terrain(data)