3.1. If Statements: 3.1.1. Simple Conditions
3.1. If Statements: 3.1.1. Simple Conditions
3.1. If Statements
3.1.1. Simple Conditions
The statements introduced in this chapter will involve tests or conditions.
More syntax for conditions will be introduced later, but for now consider
simple arithmetic comparisons that directly translate from math into Python.
Try each line separately in the Shell
2 < 5
3 > 7
x = 11
x > 10
2 * x < x
type(True)
You see that conditions are either True or False (with no quotes!). These are the
only possible Boolean values (named after 19th century mathematician
George Boole). In Python the name Boolean is shortened to the type bool . It is
the type of the results of true-false conditions or tests.
The middle two line are an if statement. It reads pretty much like English. If it
is true that the weight is greater than 50, then print the statement about an
extra charge. If it is not true that the weight is greater than 50, then dont do
the indented part: skip printing the extra luggage charge. In any event, when
you have nished with the if statement (whether it actually does anything or
not), go on to the next statement that is not indented under the if . In this
case that is the statement printing Thank you.
if condition :
indentedStatementBlock
If the condition is true, then do the indented statements. If the condition is not
true, then skip the indented statements.
if balance < 0:
transfer = -balance
# transfer enough from the backup account:
backupAccount = backupAccount - transfer
balance = balance + transfer
As with other kinds of statements with a heading and an indented block, the
block can have more than one statement. The assumption in the example
above is that if an account goes negative, it is brought back to 0 by
transferring money from a backup account in several steps.
In the examples above the choice is between doing something (if the
condition is True ) or nothing (if the condition is False ). Often there is a choice of
two possibilities, only one of which will be done, depending on the truth of a
condition.
The middle four lines are an if-else statement. Again it is close to English,
though you might say otherwise instead of else (but else is shorter!).
There are two indented blocks: One, like in the simple if statement, comes
right after the if heading and is executed when the condition in the if
heading is true. In the if - else form this is followed by an else: line, followed by
another indented block that is only executed when the original condition is
false. In an if - else statement exactly one of two possible indented blocks is
executed.
if condition :
indentedStatementBlockForTrueCondition
else:
indentedStatementBlockForFalseCondition
These statement blocks can have any number of statements, and can include
about any kind of statement.
Math
Meaning Symbol Python Symbols
Less than < <
Equals = ==
Not equal !=
Notice that the obvious choice for equals, a single equal sign, is not used to
check for equality. An annoying second equal sign is required. This is because
the single equal sign is already used for assignment in Python, so it is not
available for tests.
Warning: It is a common error to use only one equal sign when you mean
Tests for equality do not make an assignment, and they do not require a
variable on the left. Any expressions can be tested for equality or inequality
(!=). They do not need to be numbers! Predict the results and try each line in
the Shell:
x = 5
x
x == 5
x == 6
x
x != 6
x = 6
6 == x
6 != x
'hi' == 'h' + 'i'
'HI' != 'hi'
[1, 2] != [2, 1]
An equality check does not make an assignment. Strings are case sensitive.
Order matters in a list.
'a' > 5
When the comparison does not make sense, an Exception is caused. [1]
Here is another example: Pay with Overtime. Given a persons work hours for
the week and regular hourly wage, calculate the total pay for the week, taking
into account overtime. Hours worked over 40 are overtime, paid at 1.5 times
the normal rate. This is a natural place for a function enclosing the
calculation.
The problem clearly indicates two cases: when no more than 40 hours are
worked or when more than 40 hours are worked. In case more than 40 hours
are worked, it is convenient to introduce a variable overtimeHours. You are
encouraged to think about a solution before going on and examining mine.
You can try running my complete example program, wages.py, also shown
below. The format operation at the end of the main function uses the oating
point format (String Formats for Float Precision) to show two decimal places
for the cents in the answer:
def main():
hours = float(input('Enter hours worked: '))
wage = float(input('Enter dollars paid per hour: '))
total = calcWeeklyWages(hours, wage)
print('Wages for {hours} hours at ${wage:.2f} per hour are ${total:.2f}.'
.format(**locals()))
main()
Here the input was intended to be numeric, but it could be decimal so the
conversion from string was via float , not int .
for the formula in the if statement. There are generally a number of ways you
might solve the same problem!
In your main program have a simple repeat loop that calls flip() 10 times to
test it, so you generate a random sequence of 10 Heads and Tails .
The jump function is introduced for use in Strange Sequence Exercise, and
others after that.
def letterGrade(score):
if score >= 90:
letter = 'A'
else: # grade must be B, C, D or F
if score >= 80:
letter = 'B'
else: # grade must be C, D or F
if score >= 70:
letter = 'C'
else: # grade must D or F
if score >= 60:
letter = 'D'
else:
letter = 'F'
return letter
def letterGrade(score):
if score >= 90:
letter = 'A'
elif score >= 80:
letter = 'B'
elif score >= 70:
letter = 'C'
elif score >= 60:
letter = 'D'
else:
letter = 'F'
return letter
The most elaborate syntax for an if - elif - else statement is indicated in general
below:
if condition1 :
indentedStatementBlockForTrueCondition1
elif condition2 :
indentedStatementBlockForFirstTrueCondition2
elif condition3 :
indentedStatementBlockForFirstTrueCondition3
elif condition4 :
indentedStatementBlockForFirstTrueCondition4
else:
indentedStatementBlockForEachConditionFalse
The if , each elif , and the nal else line are all aligned. There can be any
number of elif lines, each followed by an indented block. (Three happen to be
illustrated above.) With this construction exactly one of the indented blocks is
executed. It is the one corresponding to the rst True condition, or, if all
conditions are False , it is the block after the nal else line.
A nal alternative for if statements: if - elif -.... with no else . This would mean
changing the syntax for if - elif - else above so the nal else: and the block after
it would be omitted. It is similar to the basic if statement without an else , in
that it is possible for no indented block to be executed. This happens if none
of the conditions in the tests are true.
With an else included, exactly one of the indented blocks is executed. Without
an else , at most one of the indented blocks is executed.
This if - elif statement only prints a line if there is a problem with the weight of
the suitcase.
Be sure to run your new version and test with dierent inputs that test all the
dierent paths through the program.
assumes people are paid double time for hours over 60. Hence they get paid
for at most 20 hours overtime at 1.5 times the normal rate. For example, a
person working 65 hours with a regular wage of $10 per hour would work at
$10 per hour for 40 hours, at 1.5 * $10 for 20 hours of overtime, and 2 * $10
for 5 hours of double time, for a total of
def printAllPositive(numberList):
'''Print only the positive numbers in numberList.'''
For example, suppose numberList is [3, -5, 2, -1, 0, 7] . You want to process a
list, so that suggests a for -each loop,
but a for -each loop runs the same code body for each element of the list, and
we only want
print(num)
for some of them. That seems like a major obstacle, but think closer at what
needs to happen concretely. As a human, who has eyes of amazing capacity,
you are drawn immediately to the actual correct numbers, 3, 2, and 7, but
clearly a computer doing this systematically will have to check every number.
In fact, there is a consistent action required: Every number must be tested to
see if it should be printed. This suggests an if statement, with the condition
num > 0 . Try loading into Idle and running the example program onlyPositive.py ,
whose code is shown below. It ends with a line testing the function:
def printAllPositive(numberList):
'''Print only the positive numbers in numberList.'''
for num in numberList:
if num > 0:
print(num)
Run example program bounce1.py . It has a red ball moving and bouncing
obliquely o the edges. If you watch several times, you should see that it
starts from random locations. Also you can repeat the program from the Shell
prompt after you have run the script. For instance, right after running the
program, try in the Shell
bounceBall(-3, 1)
The parameters give the amount the shape moves in each animation step.
You can try other values in the Shell, preferably with magnitudes less than 10.
For the remainder of the description of this example, read the extracted text
pieces.
The animations before this were totally scripted, saying exactly how many
moves in which direction, but in this case the direction of motion changes with
every bounce. The program has a graphic object shape and the central
animation step is
shape.move(dx, dy)
but in this case, dx and dy have to change when the ball gets to a boundary.
For instance, imagine the ball getting to the left side as it is moving to the left
and up. The bounce obviously alters the horizontal part of the motion, in fact
reversing it, but the ball would still continue up. The reversal of the horizontal
part of the motion means that the horizontal shift changes direction and
therefore its sign:
dx = -dx
but dy does not need to change. This switch does not happen at each
animation step, but only when the ball reaches the edge of the window. It
happens only some of the time - suggesting an if statement. Still the
condition must be determined. Suppose the center of the ball has coordinates
(x, y). When x reaches some particular x coordinate, call it xLow, the ball
should bounce.
The edge of the window is at coordinate 0, but xLow should not be 0, or the ball
would be half way o the screen before bouncing! For the edge of the ball to
hit the edge of the screen, the x coordinate of the center must be the length
of the radius away, so actually xLow is the radius of the ball.
Animation goes quickly in small steps, so I cheat. I allow the ball to take one
(small, quick) step past where it really should go ( xLow ), and then we reverse it
so it comes back to where it belongs. In particular
if x < xLow:
dx = -dx
There are similar bounding variables xHigh , yLow and yHigh , all the radius away
from the actual edge coordinates, and similar conditions to test for a bounce
o each possible edge. Note that whichever edge is hit, one coordinate, either
dx or dy, reverses. One way the collection of tests could be written is
if x < xLow:
dx = -dx
if x > xHigh:
dx = -dx
if y < yLow:
dy = -dy
if y > yHigh:
dy = -dy
This approach would cause there to be some extra testing: If it is true that x <
xLow , then it is impossible for it to be true that x > xHigh , so we do not need both
tests together. We avoid unnecessary tests with an elif clause (for both x and
y):
if x < xLow:
dx = -dx
elif x > xHigh:
dx = -dx
if y < yLow:
dy = -dy
elif y > yHigh:
dy = -dy
Note that the middle if is not changed to an elif , because it is possible for the
ball to reach a corner, and need both dx and dy reversed.
The program also uses several accessor methods for graphics objects that we
have not used in examples yet. Various graphics objects, like the circle we are
using as the shape, know their center point, and it can be accessed with the
getCenter() method. (Actually a clone of the point is returned.) Also each
coordinate of a Point can be accessed with the getX() and getY() methods.
This explains the new features in the central function dened for bouncing
around in a box, bounceInBox . The animation arbitrarily goes on in a simple
repeat loop for 600 steps. (A later example will improve this behavior.)
The program starts the ball from an arbitrary point inside the allowable
rectangular bounds. This is encapsulated in a utility function included in the
program, getRandomPoint . The getRandomPoint function uses the randrange
function from the module random . Note that in parameters for both the
functions range and randrange , the end stated is past the last value actually
desired:
The full program is listed below, repeating bounceInBox and getRandomPoint for
completeness. Several parts that may be useful later, or are easiest to follow
as a unit, are separated out as functions. Make sure you see how it all hangs
together or ask questions!
'''
Show a ball bouncing off the sides of the window.
'''
radius = 10
xLow = radius # center is separated from the wall by the radius at a bounce
xHigh = win.getWidth() - radius
yLow = radius
yHigh = win.getHeight() - radius
bounceBall(3, 5)
def printShort(strings):
'''Given a list of strings,
print the ones with at most three characters.
>>> printShort(['a', 'long', one'])
a
one
'''
In your main program, test the function, calling it several times with dierent
lists of strings. Hint: Find the length of each string with the len function.
def printEven(nums):
'''Given a list of integers nums,
print the even ones.
In your main program, test the function, calling it several times with dierent
lists of integers. Hint: A number is even if its remainder, when dividing by 2, is
0.
def chooseEven(nums):
'''Given a list of integers, nums,
return a list containing only the even ones.
In your main program, test the function, calling it several times with dierent
lists of integers and printing the results. Hint: Create a new list, and append
the appropriate numbers to it.
def uniqueList(aList):
''' Return a new list that includes the first occurrence of each value
in aList, and omits later repeats. The returned list should include
the first occurrences of values in aList in their original order.
>>> vals = ['cat', 'dog', 'cat', 'bug', 'dog', 'ant', 'dog', 'bug']
>>> uniqueList(vals)
['cat', 'dog', 'bug', 'ant']
'''
item in sequence
item not in sequence
Hint: Process aList in order. Use the new syntax to only append elements to a
new list that are not already in the new list.
After perfecting the uniqueList function, replace the last line of getKeys , so it
uses uniqueList to remove duplicates in keyList .
Check that your madlib2a.py prompts you for cue values in the order that the
cues rst appear in the madlib format string.
This is true if both credits >= 120 is true and GPA >= 2.0 is true. A short example
program using this would be:
The compound condition is true if both of the component conditions are true.
In the last example in the previous section, there was an if - elif statement
where both tests had the same block to be done if the condition was true:
if x < xLow:
dx = -dx
elif x > xHigh:
dx = -dx
There is a simpler way to state this in a sentence: If x < xLow or x > xHigh,
switch the sign of dx. That translates directly into Python:
condition1 or condition2
is true if at least one of the conditions is true. It is false if both conditions are
false. This corresponds to one way the word or is used in English. Other
times in English or is used to mean exactly one alternative is true.
that recover those two corner points, getP1 and getP2 . The program calls the
points obtained this way pt1 and pt2 . The x and y coordinates of pt1 , pt2 , and
point can be recovered with the methods of the Point type, getX() and getY() .
Suppose that I introduce variables for the x coordinates of pt1 , point , and pt2 ,
calling these x-coordinates end1 , val , and end2 , respectively. On rst try you
might decide that the needed mathematical relationship to test is
Unfortunately, this is not enough: The only requirement for the two corner
points is that they be diagonally opposite, not that the coordinates of the
second point are higher than the corresponding coordinates of the rst point.
It could be that end1 is 200; end2 is 100, and val is 120. In this latter case val is
between end1 and end2 , but substituting into the expression above
is False. The 100 and 200 need to be reversed in this case. This makes a
complicated situation. Also this is an issue which must be revisited for both
the x and y coordinates. I introduce an auxiliary function isBetween to deal with
one coordinate at a time. It starts:
Clearly this is true if the original expression, end1 <= val <= end2 , is true. You
must also consider the possible case when the order of the ends is reversed:
end2 <= val <= end1 . How do we combine these two possibilities? The Boolean
connectives to consider are and and or . Which applies? You only need one to
be true, so or is the proper connective:
if end1 <= val <= end2 or end2 <= val <= end1:
return True
else:
return False
Check the meaning: if the compound expression is True , return True . If the
condition is False , return False - in either case return the same value as the test
condition. See that a much simpler and neater version is to just return the
value of the condition itself!
return end1 <= val <= end2 or end2 <= val <= end1
between true and false values! Operate directly on the boolean expression.
Other than the two-character operators, this is like standard math syntax,
chaining comparisons. In Python any number of comparisons can be chained
in this way, closely approximating mathematical notation. Though this is good
Python, be aware that if you try other high-level languages like Java and C++,
such an expression is gibberish. Another way the expression can be expressed
(and which translates directly to other languages) is:
So much for the auxiliary function isBetween . Back to the isInside function. You
can use the isBetween function to check the x coordinates,
Again the question arises: how do you combine the two tests?
In this case we need the point to be both between the sides and between the
top and bottom, so the proper connector is and.
Sometimes you want to test the opposite of a condition. As in English you can
use the word not . For instance, to test if a Point was not inside Rectangle Rect,
you could use the condition
In general,
not condition
The program includes the functions isBetween and isInside that have already
been discussed. The program creates a number of colored rectangles to use
as buttons and also as picture components. Aside from specic data values,
the code to create each rectangle is the same, so the action is encapsulated
in a function, makeColoredRect . All of this is ne, and will be preserved in later
versions.
The present main function is long, though. It has the usual graphics starting
code, draws buttons and picture elements, and then has a number of code
sections prompting the user to choose a color for a picture element. Each
code section has a long if - elif - else test to see which button was clicked, and
sets the color of the picture element appropriately.
pt1 = rect.getP1()
pt2 = rect.getP2()
return isBetween(point.getX(), pt1.getX(), pt2.getX()) and \
isBetween(point.getY(), pt1.getY(), pt2.getY())
corner2 = corner.clone()
corner2.move(width, -height)
rect = Rectangle(corner, corner2)
rect.setFill(color)
rect.draw(win)
return rect
def main():
win = GraphWin('pick Colors', 400, 400)
win.yUp() # right side up coordinates
if isInside(pt, redButton):
color = 'red'
elif isInside(pt, yellowButton):
color = 'yellow'
elif isInside(pt, blueButton):
color = 'blue'
else :
color = 'white'
house.setFill(color)
if isInside(pt, redButton):
color = 'red'
elif isInside(pt, yellowButton):
color = 'yellow'
elif isInside(pt, blueButton):
color = 'blue'
else :
color = 'white'
door.setFill(color)
win.promptClose(msg)
main()
The only further new feature used is in the long return statement in isInside .
creator of Python decided it was best to make the syntax simple in the most
common situation. (Many other languages require a special statement
terminator symbol like ; and pay no attention to newlines). Extra
parentheses here would not hurt, so an alternative would be
s .startswith( pre )
returns Trueif string s starts with string pre: Both '-123'.startswith('-') and
'downstairs'.startswith('down') are True , but '1 - 2 - 3'.startswith('-') is False .
s .endswith( sux )
returns a new string with up to the rst count occurrences of string sub
replaced by replacement. The replacement can be the empty string to
delete sub. For example:
s = '-123'
t = s.replace('-', '', 1) # t equals '123'
t = t.replace('-', '', 1) # t is still equal to '123'
u = '.2.3.4.'
v = u.replace('.', '', 2) # v equals '23.4.'
def startsWithArticle(title):
'''Return True if the first word of title is "The", "A" or "An".'''
Be careful, if the title starts with There, it does not start with an article.
What should you be testing for?
A legal whole number string consists entirely of digits. Luckily strings have an
isdigit method, which is true when a nonempty string consists entirely of
In both parts be sure to test carefully. Not only conrm that all appropriate
strings return True . Also be sure to test that you return False for all sorts of bad
strings.
the count method from Object Orientation, and some methods from this
section.