python for-loop with 2 counters

Question

I'd like to create a single for loop using 2 variables, instead of a for loop and an external variable.

Is there a way to do something unpacking tuples with range?

Here is what I have:

space = height
for h in range(height):
    # code using both h & space

Here is the code I'm trying to improve:

# Get positive integer between 1 - 8
while True:
    height = int(input("Height: "))
    if height > 0 and height < 9:
        break

space = height  # Control space count

# Build the pyramid
for h in range(height):
    print(" " * (space - 1),"#" * (h + 1), end="")
    print(" ", end="")
    print( "#" * (h + 1), " " * (space - 1), end="")
    space -= 1

    print()  # Get prompt on \n

Since space depends decreases as h increase, why don't you just compute the offset, e.g. as height - h -1? — MisterMiyagi
– MisterMiyagi, Commented May 4, 2020 at 16:57
@MisterMiyagi - Interesting, I see how that could work. Then there would be just one variable. Thanks for pointing that out. — Wolftales
– Wolftales, Commented May 4, 2020 at 17:45
@agastya Yes, it is. I wanted to do something like C using 2 variables in a single loop. — Wolftales
– Wolftales, Commented May 4, 2020 at 17:54

DeepSpace · Accepted Answer · 2020-05-04 16:57:56Z

4

You can use a second range object (from height to 0) and then zip to iterate both ranges at once:

# Get positive integer between 1 - 8
while True:
    height = int(input("Height: "))
    if height > 0 and height < 9:
        break

# Build the pyramid
for h, space in zip(range(height), range(height, 0, -1)):
    print(" " * (space - 1),"#" * (h + 1), end="")
    print(" ", end="")
    print( "#" * (h + 1), " " * (space - 1), end="")

    print()  # Get prompt on \n

edited May 4, 2020 at 16:57

answered May 4, 2020 at 16:55

DeepSpace

82.1k12 gold badges119 silver badges166 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Matthias Over a year ago

OK, 25 seconds faster than my same solution. Will delete it now.

Wolftales Over a year ago

I figured it should be possible to construct something like this, but I was not familiar with zip(). Thank you for pointing this out!

Collectives™ on Stack Overflow

python for-loop with 2 counters

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related