author | Santosh G. Vattam <vattam.santosh@gmail.com> |
Tue, 30 Mar 2010 19:40:06 +0530 | |
changeset 9 | 538f59bb598c |
parent 5 | 76fe6a48386f |
child 11 | eafc653206d8 |
permissions | -rw-r--r-- |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
1 |
* Script |
5 | 2 |
********** |
3 |
Some greeting - Hi or Hello or Welcome - would be polite to start with |
|
4 |
********** |
|
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
5 |
|
5 | 6 |
Hello, in this tutorial, we will cover the basics of the Plotting features available in Python. We shall use Ipython and pylab. Ipython is An Enhanced Interactive Python interpreter. It provides additional features like tab completion, help etc. pylab is python library which provides plotting functionality. |
4
4dee50d4804b
Added changes to array.org file and basic plotting.
Shantanu <shantanu@fossee.in>
parents:
0
diff
changeset
|
7 |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
8 |
I am assuming that you have both of them installed on your system. |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
9 |
|
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
10 |
$ ipython -pylab |
4
4dee50d4804b
Added changes to array.org file and basic plotting.
Shantanu <shantanu@fossee.in>
parents:
0
diff
changeset
|
11 |
press RETURN |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
12 |
|
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
13 |
This will give us a prompt where we can get started. |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
14 |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
15 |
First, we create a sequence of equally spaced points starting from 0 to 2*pi |
5 | 16 |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
17 |
In []: x = lins<Tab> will automatically complete the function. This is one of the many useful features of IPython. |
5 | 18 |
|
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
19 |
In []: x = linspace(0, 2*pi, 100) |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
20 |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
21 |
To know more about the 'linspace' function type |
5 | 22 |
|
23 |
In []: lins<Tab>pace? |
|
24 |
||
25 |
It shows documentation related to linspace function. 'help' talks in detail about arguments to be passed, return values, some examples on usage. (To scroll down the page use 'SPACE' key and to scroll up use 'b')To navigate through content use arrow(/Page Up and Page Down) keys. ':See Also' section hints about other related or similar functions which might be useful. To exit help (mode) press 'q'. |
|
26 |
||
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
27 |
In this case, we have passed three arguments to the linspace function - the starting point, the last point and the total number of points. |
5 | 28 |
Check value of x by |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
29 |
In []: x |
5 | 30 |
x is a sequence of 100 points starting from 0 to 2*pi. Length of x can be seen via function |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
31 |
In []: len(x) |
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
32 |
which shows it to be 100 points. |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
33 |
|
5 | 34 |
To obtain the plot we say, |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
35 |
In []: plot(x, sin(x)) |
5 | 36 |
*** |
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
37 |
As you can see a plot has appeared on the screen. |
5 | 38 |
*** |
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
39 |
The plot has the default color and line properties. |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
40 |
|
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
41 |
Both 'pi' and 'sin' come from 'pylab'. |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
42 |
|
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
43 |
Now that we have a basic plot, we can label and title the plot. |
5 | 44 |
In []: xla<TAB>bel('x') will add a label to the x-axis. Note that 'x' is enclosed in quotes. |
45 |
Similarly |
|
46 |
In []: ylabel('sin(x)') adds a label to the y-axis. |
|
47 |
To add a title to plot we simply use |
|
48 |
In []: tit<TAB>le('Sinusoid'). |
|
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
49 |
|
5 | 50 |
Now we will add a legend to the plot. |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
51 |
In []: legend(['sin(x)']) |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
52 |
|
5 | 53 |
To go to previous command, we can use 'UP Arrow key' and 'DOWN' will take us (in reverse order)/back. |
54 |
We can modify previous command to specify the location of the legend, by passing an additional argument to the function. |
|
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
55 |
In []: legend(['sin(2y)'], loc = 'center') |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
56 |
|
5 | 57 |
other positions which can be tried are |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
58 |
'best' |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
59 |
'right' |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
60 |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
61 |
We now annotate, i.e add a comment, at a specific position in the graph. In this case, let's add a comment at the point with maximum sin value. |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
62 |
In []: annotate('local max', xy=(1.5, 1)) |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
63 |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
64 |
The first argument is the comment string and second one is the position for it. |
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
65 |
|
5 | 66 |
Now, we save the plot as follows |
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
67 |
In []: savefig('sin.png') saves the figure with the name 'sin.png' in the current directory. |
5 | 68 |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
69 |
#other supported formats are: eps, ps, pdf etc. |
5 | 70 |
|
71 |
When we use plot again by default plots get overlaid. |
|
72 |
In []: plot(x, cos(x)) |
|
73 |
||
74 |
we update Y axis label |
|
75 |
In []: ylabel('f(x)') |
|
76 |
||
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
77 |
Now in these situations with overlaid graphs, legend becomes absolutely essential. To add multiple legends, we pass the strings within quotes separated by commas and enclosed within square brackets as shown. |
5 | 78 |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
79 |
In []: legend( [ 'sin(x)' , 'cos(x)'] ) |
5 | 80 |
|
81 |
In []: clf() |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
82 |
clears the plot area and starts afresh. |
5 | 83 |
|
84 |
In case we want to create multiple plots rather than overlaid plots, we use 'figure' function. |
|
85 |
The figure command is used to open a plain figure window without any plot. |
|
86 |
In []: figure(1) |
|
87 |
||
88 |
plot() plot command plots a sin plot on figure(1) |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
89 |
In []: plot(x, sin(x)) |
5 | 90 |
|
91 |
to creates a new plain figure window without any plot. |
|
92 |
In []: figure(2) |
|
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
93 |
figure() is also used to shift the focus between multiple windows. |
5 | 94 |
|
95 |
Any command issued henceforth applies to this window only. |
|
96 |
In []: plot(x, cos(x)) |
|
97 |
The previous plot window remains unchanged to these commands. |
|
98 |
||
99 |
In []: savefig('cosine.png') |
|
100 |
||
101 |
figure(1) shifts the focus back to figure(1). |
|
102 |
In []: figure(1) |
|
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
103 |
|
5 | 104 |
title() sets the title of figure(1) |
105 |
In []: title('sin(y)') |
|
106 |
||
107 |
Here we save the plot of figure(1). |
|
108 |
In []: savefig('sine.png') |
|
109 |
||
110 |
close() closes figure(1). Now there is just one figure that is open and hence |
|
111 |
the focus is automatically shifted to figure(2). |
|
112 |
In []: close() |
|
113 |
||
114 |
close() now closes the figure(2). |
|
115 |
In []: close() |
|
116 |
||
117 |
The plot command takes the following optional parameters such as 'r' which generates the plot in red color. |
|
118 |
Use up arrow key to get till this command |
|
119 |
In []: plot(x, cos(x), 'r') and add argument. |
|
120 |
||
121 |
In []: clf() |
|
122 |
||
123 |
Passing the linewidth=2 option to plot, generates the plot with linewidth of two units. |
|
124 |
In []: plot(x, sin(x), 'g', linewidth=2) |
|
125 |
||
126 |
In []: clf() |
|
127 |
||
9
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
128 |
In order to plot points you may pass '.' as a parameter to plot |
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
129 |
In []: plot(x, sin(x), '.') |
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
130 |
|
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
131 |
In []: clf() |
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
132 |
|
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
133 |
In order to plot filled circles in black color you can pass 'ko' parameter to plot |
538f59bb598c
Minor edits to basic-plot.txt.
Santosh G. Vattam <vattam.santosh@gmail.com>
parents:
5
diff
changeset
|
134 |
In []: plot(x, cos(x), 'ko') |
5 | 135 |
|
136 |
In []: clf() |
|
137 |
||
138 |
A plot using dashed lines can be generated by passing the '--' parameter |
|
139 |
In []: plot(x, y, '--') |
|
140 |
||
141 |
You may look at more options related to colors and type of lines using plot? |
|
142 |
||
143 |
In []: clf() |
|
0
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
144 |
|
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
145 |
and finally to close the plot |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
146 |
In []: close() |
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
147 |
|
67604aed10e0
Initialization and scripts for first two sessions.
shantanu <shantanu@fossee.in>
parents:
diff
changeset
|
148 |
**************** |
5 | 149 |
This brings us to the end of this tutorial. This tutorial is first in the series of Python for Scientific Computing Tutorials. |
150 |
**************** |