This notebook is a very quick introduction to Python and in particular its scientific ecosystem in case you have never seen it before. It furthermore grants a possibility to get to know the IPython/Jupyter notebook. See here for the official documentation of the Jupyter notebook - a ton more information can be found online.
A lot of motivational writing on Why Python? is out there so we will not repeat it here and just condense it to a single sentence: Python is a good and easy to learn, open-source, general purpose programming language that happens to be very good for many scientific tasks (due to its vast scientific ecosystem).
Shift + Enter
: Execute cell and jump to the next cellCtrl/Cmd + Enter
: Execute cell and don't jump to the next cellThe tutorials are employing Jupyter notebooks but these are only one way of using Python. Writing scripts to text files and executing them with the Python interpreter of course also works:
$ python do_something.py
Another alternative is interactive usage on the command line:
$ ipython
First things first: In many notebooks you will find a cell similar to the following one. Always execute it! They do a couple of things:
print("Hello")
This essentially makes the notebooks work under Python 2 and Python 3.
# Plots now appear in the notebook.
%matplotlib inline
import matplotlib.pyplot as plt
plt.style.use('ggplot') # Matplotlib style sheet - nicer plots!
plt.rcParams['figure.figsize'] = 12, 8 # Slightly bigger plots by default
Here is collection of resources regarding the scientific Python ecosystem. They cover a number of different packages and topics; way more than we will manage today.
If you have any question regarding some specific Python functionality you can consult the official Python documenation.
Furthermore a large number of Python tutorials, introductions, and books are available online. Here are some examples for those interested in learning more.
Some people might be used to Matlab - this helps:
Additionally there is an abundance of resources introducing and teaching parts of the scientific Python ecosystem.
You might eventually have a need to create some custom plots. The quickest way to success is usually to start from some example that is somewhat similar to what you want to achieve and just modify it. These websites are good starting points:
This course is fairly non-interactive and serves to get you up to speed with Python assuming you have practical programming experience with at least one other language. Nonetheless please change things and play around an your own - it is the only way to really learn it!
The first part will introduce you to the core Python language. This tutorial uses Python 3 but almost all things can be transferred to Python 2. If possible choose Python 3 for your own work!
Python is dynamically typed and assigning something to a variable will give it that type.
# Three basic types of numbers
a = 1 # Integers
b = 2.0 # Floating Point Numbers
c = 3.0 + 4j # Complex Numbers, note the use of j for the complex part
# Arithmetics work as expected.
# Upcasting from int -> float -> complex
d = a + b # (int + float = float)
print(d)
e = c ** 2 # c to the second power, performs a complex multiplication
print(e)
Just enclose something in single or double quotes and it will become a string. On Python 3 it defaults to unicode strings, e.g. non Latin alphabets and other symbols.
# You can use single or double quotes to create strings.
location = "New York"
# Concatenate strings with plus.
where_am_i = 'I am in ' + location
# Print things with the print() function.
print(location, 1, 2)
print(where_am_i)
# Strings have a lot of attached methods for common manipulations.
print(location.lower())
# Access single items with square bracket. Negative indices are from the back.
print(location[0], location[-1])
# Strings can also be sliced.
print(location[4:])
Save your name in all lower-case letters to a variable, and print a capitalized version of it. Protip: Google for "How to capitalize a string in python". This works for almost any programming problem - someone will have had the same issue before!
name = "lion"
print(name.capitalize())
Python has two main collection types: List and dictionaries. The former is just an ordered collection of objects and is introduced here.
# List use square brackets and are simple ordered collections of things.
everything = [a, b, c, 1, 2, 3, "hello"]
# Access elements with the same slicing/indexing notation as strings.
# Note that Python indices are zero based!
print(everything[0])
print(everything[:3])
print(everything[2:-2])
# Negative indices are counted from the back of the list.
print(everything[-3:])
# Append things with the append method.
everything.append("you")
print(everything)
The other main collection type in Python are dictionaries. They are similiar to associative arrays or (hash) maps in other languages. Each entry is a key-value pair.
# Dictionaries have named fields and no inherent order. As is
# the case with lists, they can contain anything.
information = {
"name": "Hans",
"surname": "Mustermann",
"age": 78,
"kids": [1, 2, 3]
}
# Acccess items by using the key in square brackets.
print(information["kids"])
# Add new things by just assigning to a key.
print(information)
information["music"] = "jazz"
print(information)
# Delete things by using the del operator
del information["age"]
print(information)
The key to conquer a big problem is to divide it into many smaller ones and tackle them one by one. This is usually achieved by using functions.
# Functions are defined using the def keyword.
def do_stuff(a, b):
return a * b
# And called with the arguments in round brackets.
print(do_stuff(2, 3))
# Python function also can have optional arguments.
def do_more_stuff(a, b, power=1):
return (a * b) ** power
print(do_more_stuff(2, 3))
print(do_more_stuff(2, 3, power=3))
# For more complex function it is oftentimes a good idea to
#explicitly name the arguments. This is easier to read and less error-prone.
print(do_more_stuff(a=2, b=3, power=3))
To use functions and objects not part of the default namespace, you have import them. You will have to do this a lot so it is necessary to learn how to do it.
# Import anything, and use it with the dot accessor.
import math
a = math.cos(4 * math.pi)
# You can also selectively import things.
from math import pi
b = 3 * pi
# And even rename them if you don't like their name.
from math import cos as cosine
c = cosine(b)
How to know what is available?
print(dir(math))
Typing the dot and the TAB will kick off tab-completion.
math.
In the IPython framework you can also use a question mark to view the documentation of modules and functions.
math.cos?
Loops and conditionals are needed for any non-trivial task. Please note that whitespace matters in Python. Everything that is indented at the same level is part of the same block. By far the most common loops in Python are for-each loops as shown in the following. While loops also exist but are rarely used.
temp = ["a", "b", "c"]
# The typical Python loop is a for-each loop, e.g.
for item in temp:
# Everything with the same indentation is part of the loop.
new_item = item + " " + item
print(new_item)
print("No more part of the loop.")
# Useful to know is the range() function.
for i in range(5):
print(i)
The second crucial control flow structure are if/else conditional and they work the same as in any other language.
# If/else works as expected.
age = 77
if age >= 0 and age < 10:
print("Younger than ten.")
elif age >= 10:
print("Older than ten.")
else:
print("Wait what?")
# List comprehensions are a nice way to write compact loops.
# Make sure you understand this as it is very common in Python.
a = list(range(10))
print(a)
b = [i for i in a if not i % 2]
print(b)
# Equivalant loop for b.
b = []
for i in a:
if not i % 2:
b.append(i)
print(b)
You will eventually run into some error messages. Learn to read them! The last line is often the one that matters - reading upwards traces the error back in time and shows what calls led to it. If stuck: just google the error message!
def do_something(a, b):
print(a + b + something_else)
# do_something(1, 2)
The SciPy Stack forms the basis for essentially all applications of scientific Python. Here we will quickly introduce the three core libraries:
NumPy
SciPy
Matplotlib
The SciPy stack furthermore contains pandas
(library for data analysis on tabular and time series data) and sympy
(package for symbolic math), both very powerful packages, but we will omit them in this tutorial.
Large parts of the scientific Python ecosystem use NumPy, an array computation package offering N-dimensional, typed arrays and useful functions for linear algebra, Fourier transforms, random numbers, and other basic scientific tasks.
import numpy as np
# Create a large array with with 1 million samples.
x = np.linspace(start=0, stop=100, num=1E6, dtype=np.float64)
# Most operations work per-element.
y = x ** 2
# Uses C and Fortran under the hood for speed.
print(y.sum())
# FFT and inverse
x = np.random.random(100)
large_X = np.fft.fft(x)
x = np.fft.ifft(large_X)
SciPy
, in contrast to NumPy
which only offers basic numerical routines, contains a lot of additional functionality needed for scientific work. Examples are solvers for basic differential equations, numeric integration and optimization, spare matrices, interpolation routines, signal processing methods, and a lot of other things.
from scipy.interpolate import interp1d
x = np.linspace(0, 10, num=11, endpoint=True)
y = np.cos(-x ** 2 / 9.0)
# Cubic spline interpolation to new points.
f2 = interp1d(x, y, kind='cubic')(np.linspace(0, 10, num=101, endpoint=True))
Plotting is done using Matplotlib
, a package for greating high-quality static plots. It has an interface that mimics Matlab which many people are familiar with.
import matplotlib.pyplot as plt
plt.plot(np.sin(np.linspace(0, 2 * np.pi, 2000)), color="green",
label="Some Curve")
plt.legend()
plt.ylim(-1.1, 1.1)
plt.show()
import matplotlib.pyplot as plt
import numpy as np
def simple_poly(x, a, b, c):
return a * x ** 2 + b * x + c
plt.plot(simple_poly(np.linspace(-5, 5), 10, 2, 2))
plt.show()
(stolen from http://www.ling.gu.se/~lager/python_exercises.html)
"99 Bottles of Beer" is a traditional song in the United States and Canada. It is popular to sing on long trips, as it has a very repetitive format which is easy to memorize, and can take a long time to sing. The song's simple lyrics are as follows:
99 bottles of beer on the wall, 99 bottles of beer.
Take one down, pass it around, 98 bottles of beer on the wall.
The same verse is repeated, each time with one fewer bottle. The song is completed when the singer or singers reach zero.
Your task here is write a Python program capable of generating all the verses of the song.
print("99 bottles of beer on the wall, 99 bottles of beer.")
for i in range(98, -1, -1):
print("Take one down, pass it around, %i bottles of beer on the wall." % i)
(stolen from http://www.ling.gu.se/~lager/python_exercises.html)
In cryptography, a Caesar cipher is a very simple encryption techniques in which each letter in the plain text is replaced by a letter some fixed number of positions down the alphabet. For example, with a shift of 3, A would be replaced by D, B would become E, and so on. The method is named after Julius Caesar, who used it to communicate with his generals. ROT-13 ("rotate by 13 places") is a widely used example of a Caesar cipher where the shift is 13. In Python, the key for ROT-13 may be represented by means of the following dictionary:
key = {'a':'n', 'b':'o', 'c':'p', 'd':'q', 'e':'r', 'f':'s', 'g':'t', 'h':'u',
'i':'v', 'j':'w', 'k':'x', 'l':'y', 'm':'z', 'n':'a', 'o':'b', 'p':'c',
'q':'d', 'r':'e', 's':'f', 't':'g', 'u':'h', 'v':'i', 'w':'j', 'x':'k',
'y':'l', 'z':'m', 'A':'N', 'B':'O', 'C':'P', 'D':'Q', 'E':'R', 'F':'S',
'G':'T', 'H':'U', 'I':'V', 'J':'W', 'K':'X', 'L':'Y', 'M':'Z', 'N':'A',
'O':'B', 'P':'C', 'Q':'D', 'R':'E', 'S':'F', 'T':'G', 'U':'H', 'V':'I',
'W':'J', 'X':'K', 'Y':'L', 'Z':'M'}
Your task in this exercise is to implement an decoder of ROT-13. Once you're done, you will be able to read the following secret message:
Pnrfne pvcure? V zhpu cersre Pnrfne fnynq!
BONUS: Write an encoder!
sentence = "Pnrfne pvcure? V zhpu cersre Pnrfne fnynq!"
key = {'a':'n', 'b':'o', 'c':'p', 'd':'q', 'e':'r', 'f':'s', 'g':'t', 'h':'u',
'i':'v', 'j':'w', 'k':'x', 'l':'y', 'm':'z', 'n':'a', 'o':'b', 'p':'c',
'q':'d', 'r':'e', 's':'f', 't':'g', 'u':'h', 'v':'i', 'w':'j', 'x':'k',
'y':'l', 'z':'m', 'A':'N', 'B':'O', 'C':'P', 'D':'Q', 'E':'R', 'F':'S',
'G':'T', 'H':'U', 'I':'V', 'J':'W', 'K':'X', 'L':'Y', 'M':'Z', 'N':'A',
'O':'B', 'P':'C', 'Q':'D', 'R':'E', 'S':'F', 'T':'G', 'U':'H', 'V':'I',
'W':'J', 'X':'K', 'Y':'L', 'Z':'M'}
result = ""
for letter in sentence:
if letter not in key:
result += letter
else:
result += key[letter]
print(result)