9979
Comment: Revert vandalism.
|
10597
removed text saying that the code was still in Python 2 syntax
|
Deletions are marked like this. | Additions are marked like this. |
Line 1: | Line 1: |
Here are some example simple programs. Please feel free to contribute, but see notice at bottom, please. These examples assume version 2.4 or above of Python. You should be able to run them simply by copying/pasting the code into a file and running Python. Or by inserting this line (#!/bin/env python) at the beginning of your file (Unix/Linux), making the file executable (chmod u+x filename.py) and running it (./filename.py). ------ 1 line: Output {{{#!python |
Please note that these examples were changed to run under Python 3. The Python2orPython3 page provides advice on how to decide which one will best suit your needs. Some differences from Python 2 to Python 3: {{{ # Python 2 syntax |
Line 9: | Line 8: |
}}} ------ 2 lines: Input, assignment {{{#!python name = raw_input('What is your name?\n') print 'Hi, ' + name + '.' }}} 1. Usually, the `raw_input` function (renamed to `input` in Python 3) will prompt for a user's input on the same line. Writing `\n` will prompt the user on a new line. 1. The `print` command is a function in Python 3. 1. Here, we add the strings together. If we did not, and instead wrote {{{#!python start=2 numbers=off print 'Hi,',name,'.' }}} we would get {{{ Hi, Jamie . }}} Notice how there are spaces that we did not ask Python to write? These unasked for spaces occur when using commas to separate strings when using the `print` command. Rather, it may work better visually to use addition as I did in the example. ------ 3 lines: For loop, built-in enumerate function {{{#!python my_list = ['john', 'pat', 'gary', 'michael'] for i, name in enumerate(my_list): print "iteration %i is %s" % (i, name) }}} ------ 4 lines: Fibonacci, tuple assignment {{{#!python |
raw_input('What is your name?\n') self.failUnlessEqual(median([2, 9, 9, 7, 9, 2, 4, 5, 8]), 7) # Unit testing with unittest copy[(size - 1) / 2] # access of a list index without casting to int xrange(BOARD_SIZE) # xrange exists cmp(1.0, 0.0) # cmp function exists iteratable.next() # list iteration itertools.ifilter(prime.__rmod__, iteratable) # itertools.ifilter is filter in Python 3 # Python 3 syntax print('Hello, World!') input('What is your name?\n') self.assertEqual(median([2, 9, 9, 7, 9, 2, 4, 5, 8]), 7) # Unit testing with unittest, failUnlessEqual deprecated copy[int((size - 1) / 2)] # access of a list index needs casting to int range(BOARD_SIZE) # xrange was renamed to range in Python 3 #cmp needs to be implemented! def cmp(a, b): return (a > b) - (a < b) next(iteratable) # list iteration filter(prime.__rmod__, iteratable) # is itertools.ifilter in Python 2 }}} The examples below will increase in number of lines of code and difficulty: 1 line: Output {{{ print ('Hello, world!') }}} ---- 2 lines: Input, assignment {{{ name = input('What is your name?\n') print ('Hi, %s.' % name) }}} ---- 3 lines: For loop, built-in enumerate function, new style formatting {{{ friends = ['john', 'pat', 'gary', 'michael'] for i, name in enumerate(friends): print ("iteration {iteration} is {name}".format(iteration=i, name=name)) }}} ---- 4 lines: Fibonacci, tuple assignment {{{ |
Line 44: | Line 56: |
print 'This generation has %d babies' % babies | print ('This generation has {0} babies'.format(babies)) |
Line 47: | Line 59: |
------ 5 lines: Functions {{{#!python |
---- 5 lines: Functions {{{ |
Line 51: | Line 64: |
print 'hello', name | print ('Hello', name) |
Line 56: | Line 70: |
------ 6 lines: Import, regular expressions {{{#!python |
---- 6 lines: Import, regular expressions {{{ |
Line 62: | Line 77: |
print test_string, 'is a valid US local phone number' | print (test_string, 'is a valid US local phone number') |
Line 64: | Line 79: |
print test_string, 'rejected' }}} ------ 7 lines: Dictionaries, generator expressions {{{#!python |
print (test_string, 'rejected') }}} ---- 7 lines: Dictionaries, generator expressions {{{ |
Line 75: | Line 91: |
print 'I owe the grocer $%.2f' % grocery_bill }}} ------ 8 lines: Command line arguments, exception handling {{{#!python #!/usr/bin/env python # This program adds up integers in the command line |
print ('I owe the grocer $%.2f' % grocery_bill) }}} ---- 8 lines: Command line arguments, exception handling {{{ # This program adds up integers that have been passed as arguments in the command line |
Line 86: | Line 101: |
print 'sum =', total | print ('sum =', total) |
Line 88: | Line 103: |
print 'Please supply integer arguments' }}} ------ 9 lines: Opening files {{{#!python |
print ('Please supply integer arguments') }}} ---- 9 lines: Opening files {{{ |
Line 98: | Line 113: |
for fn in sorted(python_files): print ' ------', fn for line in open(fn): print ' ' + line.rstrip() }}} ------ 10 lines: Time, conditionals {{{#!python import time now = time.localtime() hour = now.tm_hour if hour < 8: print 'sleeping' elif hour < 9: print 'commuting' elif hour < 17: print 'working' elif hour < 18: print 'commuting' elif hour < 20: print 'eating' elif hour < 22: print 'resting' else: print 'sleeping' }}} ------ 11 lines: Triple-quoted strings, while loop {{{#!python |
for file_name in sorted(python_files): print (' ------' + file_name) with open(file_name) as f: for line in f: print (' ' + line.rstrip()) print() }}} ---- 10 lines: Time, conditionals, from..import, for..else {{{ from time import localtime activities = {8: 'Sleeping', 9: 'Commuting', 17: 'Working', 18: 'Commuting', 20: 'Eating', 22: 'Resting' } time_now = localtime() hour = time_now.tm_hour for activity_time in sorted(activities.keys()): if hour < activity_time: print (activities[activity_time]) break else: print ('Unknown, AFK or sleeping!') }}} ---- 11 lines: Triple-quoted strings, while loop {{{ |
Line 127: | Line 155: |
bottles_of_beer = 99 | bottles_of_beer = 9 |
Line 129: | Line 157: |
print REFRAIN % (bottles_of_beer, bottles_of_beer, bottles_of_beer - 1) |
print (REFRAIN % (bottles_of_beer, bottles_of_beer, bottles_of_beer - 1)) |
Line 133: | Line 161: |
------ 12 lines: Classes {{{#!python |
---- 12 lines: Classes {{{ |
Line 146: | Line 175: |
my_account.withdraw(5) print my_account.balance }}} ------ 13 lines: Unit testing with unittest {{{#!python |
my_account.withdraw(50) print (my_account.balance, my_account.overdrawn()) }}} ---- 13 lines: Unit testing with unittest {{{ |
Line 157: | Line 187: |
return copy[(size - 1) / 2] | return copy[int((size - 1) / 2)] |
Line 159: | Line 189: |
return (copy[size/2 - 1] + copy[size/2]) / 2 | return (copy[int(size/2 - 1)] + copy[int(size/2)]) / 2 |
Line 162: | Line 192: |
self.failUnlessEqual(median([2, 9, 9, 7, 9, 2, 4, 5, 8]), 7) | self.assertEqual(median([2, 9, 9, 7, 9, 2, 4, 5, 8]), 7) |
Line 166: | Line 196: |
------ 14 lines: Doctest-based testing {{{#!python |
---- 14 lines: Doctest-based testing {{{ |
Line 173: | Line 203: |
7 | 6 #change to 7 in order to pass the test |
Line 178: | Line 208: |
return copy[(size - 1) / 2] | return copy[int((size - 1) / 2)] |
Line 180: | Line 210: |
return (copy[size/2 - 1] + copy[size/2]) / 2 | return (copy[int(size/2 - 1)] + copy[int(size/2)]) / 2 |
Line 185: | Line 215: |
------ 15 lines: itertools {{{#!python import itertools |
---- 15 lines: itertools {{{ from itertools import groupby |
Line 197: | Line 228: |
for has_chars, frags in itertools.groupby(lines, bool): | for has_chars, frags in groupby(lines, bool): |
Line 199: | Line 230: |
print ' '.join(frags) | print (' '.join(frags)) |
Line 204: | Line 235: |
------ 16 lines: csv module, tuple unpacking, cmp() built-in {{{#!python |
---- 16 lines: csv module, tuple unpacking, cmp() built-in {{{ |
Line 209: | Line 241: |
# need to define cmp function in Python 3 def cmp(a, b): return (a > b) - (a < b) |
|
Line 210: | Line 246: |
writer = csv.writer(open('stocks.csv', 'wb', buffering=0)) writer.writerows([ ('GOOG', 'Google, Inc.', 505.24, 0.47, 0.09), ('YHOO', 'Yahoo! Inc.', 27.38, 0.33, 1.22), ('CNET', 'CNET Networks, Inc.', 8.62, -0.13, -1.49) ]) |
with open('stocks.csv', 'w', newline='') as stocksFileW: writer = csv.writer(stocksFileW) writer.writerows([ ['GOOG', 'Google, Inc.', 505.24, 0.47, 0.09], ['YHOO', 'Yahoo! Inc.', 27.38, 0.33, 1.22], ['CNET', 'CNET Networks, Inc.', 8.62, -0.13, -1.4901] ]) |
Line 218: | Line 255: |
stocks = csv.reader(open('stocks.csv', 'rb')) status_labels = {-1: 'down', 0: 'unchanged', 1: 'up'} for ticker, name, price, change, pct in stocks: status = status_labels[cmp(float(change), 0.0)] print '%s is %s (%s%%)' % (name, status, pct) }}} ------ 18 lines: 8-Queens Problem (recursion) {{{#!python |
with open('stocks.csv', 'r') as stocksFile: stocks = csv.reader(stocksFile) status_labels = {-1: 'down', 0: 'unchanged', 1: 'up'} for ticker, name, price, change, pct in stocks: status = status_labels[cmp(float(change), 0.0)] print ('%s is %s (%.2f)' % (name, status, float(pct))) }}} ---- 18 lines: 8-Queens Problem (recursion) {{{ |
Line 230: | Line 271: |
Line 231: | Line 273: |
left, right = left-1, right+1 | left, right = left - 1, right + 1 |
Line 237: | Line 280: |
if n == 0: return [[]] smaller_solutions = solve(n-1) |
if n == 0: return [[]] smaller_solutions = solve(n - 1) |
Line 243: | Line 289: |
for answer in solve(BOARD_SIZE): print answer }}} ------ 20 lines: Prime numbers sieve w/fancy generators {{{#!python |
for answer in solve(BOARD_SIZE): print (answer) }}} ---- 20 lines: Prime numbers sieve w/fancy generators {{{ |
Line 257: | Line 305: |
prime = numbers.next() | prime = next(numbers) |
Line 262: | Line 310: |
numbers = itertools.ifilter(prime.__rmod__, numbers) | numbers = filter(prime.__rmod__, numbers) |
Line 267: | Line 315: |
print p }}} ------ 21 lines: XML/HTML parsing (using Python 2.5 or third-party library) {{{#!python |
print (p) }}} ---- 21 lines: XML/HTML parsing {{{ |
Line 280: | Line 329: |
# In Python 2.5 or from http://effbot.org/zone/element-index.htm | # From http://effbot.org/zone/element-index.htm |
Line 292: | Line 341: |
print "%s: %s %s" % (item.text, amt.text, unit.text) }}} ------ 28 lines: 8-Queens Problem (define your own exceptions) {{{#!python |
print ("%s: %s %s" % (item.text, amt.text, unit.text)) }}} ---- 28 lines: 8-Queens Problem (define your own exceptions) {{{ |
Line 323: | Line 373: |
print queens print "\n".join(". "*q + "Q " + ". "*(BOARD_SIZE-q-1) for q in queens) }}} ------ Hi, I started this page in May 2007, and I provided the first 10+ or so examples (which may have changed since then). -- SteveHowell All code on this page is open source, of course, with the standard Python license. Minor cleanups are welcome, but if you want to do major restructuring of this page, please run them by the folks on the Python mailing list, or if you are impatient for a response, please just make your own copy of this page. Thanks, and I hope this code is useful for you! Some goals for this page: 1. All examples should be simple. 2. There should be a gentle progression through Python concepts. |
print (queens) print ("\n".join(". "*q + "Q " + ". "*(BOARD_SIZE-q-1) for q in queens)) }}} ---- 33 lines: "Guess the Number" Game (edited) from http://inventwithpython.com {{{ import random guesses_made = 0 name = input('Hello! What is your name?\n') number = random.randint(1, 20) print ('Well, {0}, I am thinking of a number between 1 and 20.'.format(name)) while guesses_made < 6: guess = int(input('Take a guess: ')) guesses_made += 1 if guess < number: print ('Your guess is too low.') if guess > number: print ('Your guess is too high.') if guess == number: break if guess == number: print ('Good job, {0}! You guessed my number in {1} guesses!'.format(name, guesses_made)) else: print ('Nope. The number I was thinking of was {0}'.format(number)) }}} |
Please note that these examples were changed to run under Python 3. The Python2orPython3 page provides advice on how to decide which one will best suit your needs.
Some differences from Python 2 to Python 3:
# Python 2 syntax print 'Hello, world!' raw_input('What is your name?\n') self.failUnlessEqual(median([2, 9, 9, 7, 9, 2, 4, 5, 8]), 7) # Unit testing with unittest copy[(size - 1) / 2] # access of a list index without casting to int xrange(BOARD_SIZE) # xrange exists cmp(1.0, 0.0) # cmp function exists iteratable.next() # list iteration itertools.ifilter(prime.__rmod__, iteratable) # itertools.ifilter is filter in Python 3 # Python 3 syntax print('Hello, World!') input('What is your name?\n') self.assertEqual(median([2, 9, 9, 7, 9, 2, 4, 5, 8]), 7) # Unit testing with unittest, failUnlessEqual deprecated copy[int((size - 1) / 2)] # access of a list index needs casting to int range(BOARD_SIZE) # xrange was renamed to range in Python 3 #cmp needs to be implemented! def cmp(a, b): return (a > b) - (a < b) next(iteratable) # list iteration filter(prime.__rmod__, iteratable) # is itertools.ifilter in Python 2
The examples below will increase in number of lines of code and difficulty:
1 line: Output
print ('Hello, world!')
2 lines: Input, assignment
name = input('What is your name?\n') print ('Hi, %s.' % name)
3 lines: For loop, built-in enumerate function, new style formatting
friends = ['john', 'pat', 'gary', 'michael'] for i, name in enumerate(friends): print ("iteration {iteration} is {name}".format(iteration=i, name=name))
4 lines: Fibonacci, tuple assignment
parents, babies = (1, 1) while babies < 100: print ('This generation has {0} babies'.format(babies)) parents, babies = (babies, parents + babies)
5 lines: Functions
def greet(name): print ('Hello', name) greet('Jack') greet('Jill') greet('Bob')
6 lines: Import, regular expressions
import re for test_string in ['555-1212', 'ILL-EGAL']: if re.match(r'^\d{3}-\d{4}$', test_string): print (test_string, 'is a valid US local phone number') else: print (test_string, 'rejected')
7 lines: Dictionaries, generator expressions
prices = {'apple': 0.40, 'banana': 0.50} my_purchase = { 'apple': 1, 'banana': 6} grocery_bill = sum(prices[fruit] * my_purchase[fruit] for fruit in my_purchase) print ('I owe the grocer $%.2f' % grocery_bill)
8 lines: Command line arguments, exception handling
# This program adds up integers that have been passed as arguments in the command line import sys try: total = sum(int(arg) for arg in sys.argv[1:]) print ('sum =', total) except ValueError: print ('Please supply integer arguments')
9 lines: Opening files
# indent your Python code to put into an email import glob # glob supports Unix style pathname extensions python_files = glob.glob('*.py') for file_name in sorted(python_files): print (' ------' + file_name) with open(file_name) as f: for line in f: print (' ' + line.rstrip()) print()
10 lines: Time, conditionals, from..import, for..else
from time import localtime activities = {8: 'Sleeping', 9: 'Commuting', 17: 'Working', 18: 'Commuting', 20: 'Eating', 22: 'Resting' } time_now = localtime() hour = time_now.tm_hour for activity_time in sorted(activities.keys()): if hour < activity_time: print (activities[activity_time]) break else: print ('Unknown, AFK or sleeping!')
11 lines: Triple-quoted strings, while loop
REFRAIN = ''' %d bottles of beer on the wall, %d bottles of beer, take one down, pass it around, %d bottles of beer on the wall! ''' bottles_of_beer = 9 while bottles_of_beer > 1: print (REFRAIN % (bottles_of_beer, bottles_of_beer, bottles_of_beer - 1)) bottles_of_beer -= 1
12 lines: Classes
class BankAccount(object): def __init__(self, initial_balance=0): self.balance = initial_balance def deposit(self, amount): self.balance += amount def withdraw(self, amount): self.balance -= amount def overdrawn(self): return self.balance < 0 my_account = BankAccount(15) my_account.withdraw(50) print (my_account.balance, my_account.overdrawn())
13 lines: Unit testing with unittest
import unittest def median(pool): copy = sorted(pool) size = len(copy) if size % 2 == 1: return copy[int((size - 1) / 2)] else: return (copy[int(size/2 - 1)] + copy[int(size/2)]) / 2 class TestMedian(unittest.TestCase): def testMedian(self): self.assertEqual(median([2, 9, 9, 7, 9, 2, 4, 5, 8]), 7) if __name__ == '__main__': unittest.main()
14 lines: Doctest-based testing
def median(pool): '''Statistical median to demonstrate doctest. >>> median([2, 9, 9, 7, 9, 2, 4, 5, 8]) 6 #change to 7 in order to pass the test ''' copy = sorted(pool) size = len(copy) if size % 2 == 1: return copy[int((size - 1) / 2)] else: return (copy[int(size/2 - 1)] + copy[int(size/2)]) / 2 if __name__ == '__main__': import doctest doctest.testmod()
15 lines: itertools
from itertools import groupby lines = ''' This is the first paragraph. This is the second. '''.splitlines() # Use itertools.groupby and bool to return groups of # consecutive lines that either have content or don't. for has_chars, frags in groupby(lines, bool): if has_chars: print (' '.join(frags)) # PRINTS: # This is the first paragraph. # This is the second.
16 lines: csv module, tuple unpacking, cmp() built-in
import csv # need to define cmp function in Python 3 def cmp(a, b): return (a > b) - (a < b) # write stocks data as comma-separated values with open('stocks.csv', 'w', newline='') as stocksFileW: writer = csv.writer(stocksFileW) writer.writerows([ ['GOOG', 'Google, Inc.', 505.24, 0.47, 0.09], ['YHOO', 'Yahoo! Inc.', 27.38, 0.33, 1.22], ['CNET', 'CNET Networks, Inc.', 8.62, -0.13, -1.4901] ]) # read stocks data, print status messages with open('stocks.csv', 'r') as stocksFile: stocks = csv.reader(stocksFile) status_labels = {-1: 'down', 0: 'unchanged', 1: 'up'} for ticker, name, price, change, pct in stocks: status = status_labels[cmp(float(change), 0.0)] print ('%s is %s (%.2f)' % (name, status, float(pct)))
18 lines: 8-Queens Problem (recursion)
BOARD_SIZE = 8 def under_attack(col, queens): left = right = col for r, c in reversed(queens): left, right = left - 1, right + 1 if c in (left, col, right): return True return False def solve(n): if n == 0: return [[]] smaller_solutions = solve(n - 1) return [solution+[(n,i+1)] for i in range(BOARD_SIZE) for solution in smaller_solutions if not under_attack(i+1, solution)] for answer in solve(BOARD_SIZE): print (answer)
20 lines: Prime numbers sieve w/fancy generators
import itertools def iter_primes(): # an iterator of all numbers between 2 and +infinity numbers = itertools.count(2) # generate primes forever while True: # get the first number from the iterator (always a prime) prime = next(numbers) yield prime # this code iteratively builds up a chain of # filters...slightly tricky, but ponder it a bit numbers = filter(prime.__rmod__, numbers) for p in iter_primes(): if p > 1000: break print (p)
21 lines: XML/HTML parsing
dinner_recipe = '''<html><body><table> <tr><th>amt</th><th>unit</th><th>item</th></tr> <tr><td>24</td><td>slices</td><td>baguette</td></tr> <tr><td>2+</td><td>tbsp</td><td>olive oil</td></tr> <tr><td>1</td><td>cup</td><td>tomatoes</td></tr> <tr><td>1</td><td>jar</td><td>pesto</td></tr> </table></body></html>''' # From http://effbot.org/zone/element-index.htm import xml.etree.ElementTree as etree tree = etree.fromstring(dinner_recipe) # For invalid HTML use http://effbot.org/zone/element-soup.htm # import ElementSoup, StringIO # tree = ElementSoup.parse(StringIO.StringIO(dinner_recipe)) pantry = set(['olive oil', 'pesto']) for ingredient in tree.getiterator('tr'): amt, unit, item = ingredient if item.tag == "td" and item.text not in pantry: print ("%s: %s %s" % (item.text, amt.text, unit.text))
28 lines: 8-Queens Problem (define your own exceptions)
BOARD_SIZE = 8 class BailOut(Exception): pass def validate(queens): left = right = col = queens[-1] for r in reversed(queens[:-1]): left, right = left-1, right+1 if r in (left, col, right): raise BailOut def add_queen(queens): for i in range(BOARD_SIZE): test_queens = queens + [i] try: validate(test_queens) if len(test_queens) == BOARD_SIZE: return test_queens else: return add_queen(test_queens) except BailOut: pass raise BailOut queens = add_queen([]) print (queens) print ("\n".join(". "*q + "Q " + ". "*(BOARD_SIZE-q-1) for q in queens))
33 lines: "Guess the Number" Game (edited) from http://inventwithpython.com
import random guesses_made = 0 name = input('Hello! What is your name?\n') number = random.randint(1, 20) print ('Well, {0}, I am thinking of a number between 1 and 20.'.format(name)) while guesses_made < 6: guess = int(input('Take a guess: ')) guesses_made += 1 if guess < number: print ('Your guess is too low.') if guess > number: print ('Your guess is too high.') if guess == number: break if guess == number: print ('Good job, {0}! You guessed my number in {1} guesses!'.format(name, guesses_made)) else: print ('Nope. The number I was thinking of was {0}'.format(number))