- for loops
def sumFromMToN(m, n):
total = 0
# note that range(x, y) includes x but excludes y
for x in range(m, n+1):
total += x
return total
print(sumFromMToN(5, 10) == 5+6+7+8+9+10)
What if we omit the first parameter?
def sumToN(n):
total = 0
for x in range(n+1):
total += x
return total
print(sumToN(5) == 0+1+2+3+4+5)
What if we add a third parameter?
def sumEveryKthFromMToN(m, n, k):
total = 0
for x in range(m, n+1, k):
total += x
return total
print(sumEveryKthFromMToN(5, 20, 7) == (5 + 12 + 19))
Sum just odd numbers from m to n:
def sumOfOddsFromMToN(m, n):
total = 0
for x in range(m, n+1):
if (x % 2 == 1):
total += x
return total
print(sumOfOddsFromMToN(4, 10) == sumOfOddsFromMToN(5,9) == (5+7+9))
- Nested for loops
def printCoordinates(xMax, yMax):
for x in range(xMax+1):
for y in range(yMax+1):
print("(", x, ",", y, ") ", end="")
print()
printCoordinates(4, 5)
How about some stars?
def printStarRectangle(n):
# print an nxn rectangle of asterisks
for row in range(n):
for col in range(n):
print("*", end="")
print()
printStarRectangle(5)
And again:
# What would this do? Be careful and be precise!
def printMysteryStarShape(n):
for row in range(n):
print(row, end=" ")
for col in range(row):
print("*", end=" ")
print()
printMysteryStarShape(5)
- while loops
# use while loops when there is an indeterminate number of iterations
def leftmostDigit(n):
n = abs(n)
while (n >= 10):
n = n//10
return n
print(leftmostDigit(72658489290098) == 7)
Misuse: While loop over a fixed range
# sum numbers from 1 to 10
# note: this works, but you should not use "while" here.
# instead, do this with "for" (once you know how)
def sumToN(n):
# note: even though this works, it is bad style.
# You should do this with a "for" loop, not a "while" loop.
total = 0
counter = 1
while (counter <= n):
total += counter
counter += 1
return total
print(sumToN(5) == 1+2+3+4+5)
Once again, but with a bug!:
def buggySumToN(n):
# note: this not only uses a "while" instead of a "for" loop,
# but it also contains a bug. Ugh.
total = 0
counter = 0
while (counter <= n):
counter += 1
total += counter
return total
print(buggySumToN(5) == 1+2+3+4+5)
And once again, with a for loop:
# A for loop is the preferred way to loop over a fixed range.
def sumToN(n):
total = 0
for counter in range(n+1):
total += counter
return total
print(sumToN(5) == 1+2+3+4+5)
- break and continue
for n in range(200):
if (n % 3 == 0):
continue # skips rest of this pass (rarely a good idea to use "continue")
elif (n == 8):
break # skips rest of entire loop
print(n, end=" ")
print()
Infinite "while" loop with break:
def readUntilDone():
linesEntered = 0
while (True):
response = input("Enter a string (or 'done' to quit): ")
if (response == "done"):
break
print(" You entered: ", response)
linesEntered += 1
print("Bye!")
return linesEntered
linesEntered = readUntilDone()
print("You entered", linesEntered, "lines (not counting 'done').")
- isPrime
# Note: there are faster/better ways. We're just going for clarity and simplicity here.
def isPrime(n):
if (n < 2):
return False
for factor in range(2,n):
if (n % factor == 0):
return False
return True
# And take it for a spin
for n in range(100):
if isPrime(n):
print(n, end=" ")
print()
fasterIsPrime:
#Note: this is still not the fastest way, but it's a nice improvement.
def fasterIsPrime(n):
if (n < 2):
return False
if (n == 2):
return True
if (n % 2 == 0):
return False
maxFactor = round(n**0.5)
for factor in range(3,maxFactor+1,2):
if (n % factor == 0):
return False
return True
# And try out this version:
for n in range(100):
if fasterIsPrime(n):
print(n, end=" ")
print()
- nthPrime
def isPrime(n):
if (n < 2):
return False
if (n == 2):
return True
if (n % 2 == 0):
return False
maxFactor = round(n**0.5)
for factor in range(3,maxFactor+1,2):
if (n % factor == 0):
return False
return True
def nthPrime(n):
found = 0
guess = 0
while (found <= n):
guess += 1
if (isPrime(guess)):
found += 1
return guess
# and let's see a list of the primes
for n in range(10):
print(n, nthPrime(n))
print("Done!")