01-numpy.md 33.5 KB
Newer Older
1
---
2
3
4
5
title: Analyzing Patient Data
teaching: 30
exercises: 0
questions:
Greg Wilson's avatar
Greg Wilson committed
6
- "How can I process tabular data files in Python?"
7
8
objectives:
- "Explain what a library is, and what libraries are used for."
9
- "Import a Python library and use the functions it contains."
10
11
12
13
- "Read tabular data from a file into a program."
- "Assign values to variables."
- "Select individual values and subsections from data."
- "Perform operations on arrays of data."
14
- "Plot simple graphs from data."
15
keypoints:
Greg Wilson's avatar
Greg Wilson committed
16
17
18
19
20
21
22
23
24
25
26
27
28
29
- "Import a library into a program using `import libraryname`."
- "Use the `numpy` library to work with arrays in Python."
- "Use `variable = value` to assign a value to a variable in order to record it in memory."
- "Variables are created on demand whenever a value is assigned to them."
- "Use `print(something)` to display the value of `something`."
- "The expression `array.shape` gives the shape of an array."
- "Use `array[x, y]` to select a single element from an array."
- "Array indices start at 0, not 1."
- "Use `low:high` to specify a slice that includes the indices from `low` to `high-1`."
- "All the indexing and slicing that works on arrays also works on strings."
- "Use `# some kind of explanation` to add comments to programs."
- "Use `numpy.mean(array)`, `numpy.max(array)`, and `numpy.min(array)` to calculate simple statistics."
- "Use `numpy.mean(array, axis=0)` or `numpy.mean(array, axis=1)` to calculate statistics across the specified axis."
- "Use the `pyplot` library from `matplotlib` for creating simple visualizations."
30
---
31
In this lesson we will learn how to manipulate the inflammation dataset with Python. But before we discuss how to deal with many data points, we will show how to store a single value on the computer.
32

33
The line below [assigns](reference.html#assignment) the value `55` to a [variable](reference.html#variable) `weight_kg`:
34

35
~~~
36
weight_kg = 55
Greg Wilson's avatar
Greg Wilson committed
37
~~~
38
39
{: .python}

40
A variable is just a name for a value,
Greg Wilson's avatar
Greg Wilson committed
41
such as `x`, `current_temperature`, or `subject_id`.
42
Python's variables must begin with a letter and are [case sensitive](reference.html#case-sensitive).
Kyler Brown's avatar
Kyler Brown committed
43
We can create a new variable by assigning a value to it using `=`.
44
45
When we are finished typing and press Shift+Enter,
the notebook runs our command.
46

47
Once a variable has a value, we can print it to the screen:
48

49
~~~
50
print(weight_kg)
Greg Wilson's avatar
Greg Wilson committed
51
~~~
52
53
54
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
55
56
55
~~~
57
{: .output}
58

59
and do arithmetic with it:
60

61
~~~
62
print('weight in pounds:', 2.2 * weight_kg)
Greg Wilson's avatar
Greg Wilson committed
63
~~~
64
65
66
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
67
68
weight in pounds: 121.0
~~~
69
{: .output}
70

jstapleton's avatar
jstapleton committed
71
72
73
As the example above shows,
we can print several things at once by separating them with commas.

74
We can also change a variable's value by assigning it a new one:
75

76
~~~
Greg Wilson's avatar
Greg Wilson committed
77
weight_kg = 57.5
78
print('weight in kilograms is now:', weight_kg)
Greg Wilson's avatar
Greg Wilson committed
79
~~~
80
81
82
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
83
84
weight in kilograms is now: 57.5
~~~
85
{: .output}
86

87
88
If we imagine the variable as a sticky note with a name written on it,
assignment is like putting the sticky note on a particular value:
89

90
![Variables as Sticky Notes](../fig/python-sticky-note-variables-01.svg)
91

92
93
94
This means that assigning a value to one variable does *not* change the values of other variables.
For example,
let's store the subject's weight in pounds in a variable:
95

96
~~~
Greg Wilson's avatar
Greg Wilson committed
97
weight_lb = 2.2 * weight_kg
98
print('weight in kilograms:', weight_kg, 'and in pounds:', weight_lb)
Greg Wilson's avatar
Greg Wilson committed
99
~~~
100
101
102
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
103
104
weight in kilograms: 57.5 and in pounds: 126.5
~~~
105
{: .output}
106

107
![Creating Another Variable](../fig/python-sticky-note-variables-02.svg)
108

109
and then change `weight_kg`:
110

111
~~~
Greg Wilson's avatar
Greg Wilson committed
112
weight_kg = 100.0
113
print('weight in kilograms is now:', weight_kg, 'and weight in pounds is still:', weight_lb)
Greg Wilson's avatar
Greg Wilson committed
114
~~~
115
116
117
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
118
119
weight in kilograms is now: 100.0 and weight in pounds is still: 126.5
~~~
120
{: .output}
121

122
![Updating a Variable](../fig/python-sticky-note-variables-03.svg)
123

124
125
126
127
Since `weight_lb` doesn't "remember" where its value came from,
it isn't automatically updated when `weight_kg` changes.
This is different from the way spreadsheets work.

128
> ## Who's Who in Memory
Benjamin Laken's avatar
Benjamin Laken committed
129
>
130
131
132
> You can use the `%whos` command at any time to see what
> variables you have created and what modules you have loaded into the computer's memory.
> As this is an IPython command, it will only work if you are in an IPython terminal or the Jupyter Notebook.
Benjamin Laken's avatar
Benjamin Laken committed
133
>
134
> ~~~
135
136
> %whos
> ~~~
137
138
139
> {: .python}
>
> ~~~
140
141
142
143
144
145
> Variable    Type       Data/Info
> --------------------------------
> numpy       module     <module 'numpy' from '/Us<...>kages/numpy/__init__.py'>
> weight_kg   float      100.0
> weight_lb   float      126.5
> ~~~
146
147
> {: .output}
{: .callout}
Benjamin Laken's avatar
Benjamin Laken committed
148

devendra1810's avatar
devendra1810 committed
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
Words are useful,
but what's more useful are the sentences and stories we build with them.
Similarly,
while a lot of powerful, general tools are built into languages like Python,
specialized tools built up from these basic units live in [libraries](reference.html#library)
that can be called upon when needed.

In order to load our inflammation data,
we need to access ([import](reference.html#import) in Python terminology)
a library called [NumPy](http://docs.scipy.org/doc/numpy/ "NumPy Documentation").
In general you should use this library if you want to do fancy things with numbers,
especially if you have matrices or arrays.
We can import NumPy using:

~~~
import numpy
~~~
{: .python}

Importing a library is like getting a piece of lab equipment out of a storage locker and setting it up on the bench.
Libraries provide additional functionality to the basic Python package,
170
171
much like a new piece of equipment adds functionality to a lab space. Just like in the lab, importing too many libraries
can sometimes complicate and slow down your programs - so we only import what we need for each program. 
devendra1810's avatar
devendra1810 committed
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
Once you've imported the library,
we can ask the library to read our data file for us:

~~~
numpy.loadtxt(fname='inflammation-01.csv', delimiter=',')
~~~
{: .python}

~~~
array([[ 0.,  0.,  1., ...,  3.,  0.,  0.],
       [ 0.,  1.,  2., ...,  1.,  0.,  1.],
       [ 0.,  1.,  1., ...,  2.,  1.,  1.],
       ...,
       [ 0.,  1.,  1., ...,  1.,  1.,  1.],
       [ 0.,  0.,  0., ...,  0.,  2.,  0.],
       [ 0.,  0.,  1., ...,  1.,  1.,  0.]])
~~~
{: .output}

The expression `numpy.loadtxt(...)` is a [function call](reference.html#function-call)
that asks Python to run the [function](reference.html#function) `loadtxt` which belongs to the `numpy` library.
This [dotted notation](reference.html#dotted-notation) is used everywhere in Python
to refer to the parts of things as `thing.component`.

`numpy.loadtxt` has two [parameters](reference.html#parameter):
the name of the file we want to read,
and the [delimiter](reference.html#delimiter) that separates values on a line.
These both need to be character strings (or [strings](reference.html#string) for short),
so we put them in quotes.

Since we haven't told it to do anything else with the function's output,
the notebook displays it.
In this case,
that output is the data we just loaded.
By default,
only a few rows and columns are shown
(with `...` to omit elements when displaying big arrays).
To save space,
Python displays numbers as `1.` instead of `1.0`
when there's nothing interesting after the decimal point.

Our call to `numpy.loadtxt` read our file,
but didn't save the data in memory.
To do that,
216
we need to assign the array to a variable. Just as we can assign a single value to a variable, we can also assign an array of values
Johnny Lin's avatar
Johnny Lin committed
217
to a variable using the same syntax.  Let's re-run `numpy.loadtxt` and save its result:
218

219
~~~
Greg Wilson's avatar
Greg Wilson committed
220
221
data = numpy.loadtxt(fname='inflammation-01.csv', delimiter=',')
~~~
222
{: .python}
223

224
225
226
This statement doesn't produce any output because assignment doesn't display anything.
If we want to check that our data has been loaded,
we can print the variable's value:
227

228
~~~
229
print(data)
Greg Wilson's avatar
Greg Wilson committed
230
~~~
231
232
233
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
234
[[ 0.  0.  1. ...,  3.  0.  0.]
235
236
 [ 0.  1.  2. ...,  1.  0.  1.]
 [ 0.  1.  1. ...,  2.  1.  1.]
237
 ...,
238
239
240
 [ 0.  1.  1. ...,  1.  1.  1.]
 [ 0.  0.  0. ...,  0.  2.  0.]
 [ 0.  0.  1. ...,  1.  1.  0.]]
Greg Wilson's avatar
Greg Wilson committed
241
~~~
242
{: .output}
243

244
245
246
Now that our data is in memory,
we can start doing things with it.
First,
247
let's ask what [type]({{ page.root }}/reference/#type) of thing `data` refers to:
248

249
~~~
250
print(type(data))
Greg Wilson's avatar
Greg Wilson committed
251
~~~
252
253
254
{: .python}

~~~
255
<class 'numpy.ndarray'>
Greg Wilson's avatar
Greg Wilson committed
256
~~~
257
{: .output}
258

259
260
261
262
263
264
The output tells us that `data` currently refers to
an N-dimensional array created by the NumPy library.
These data correspond to arthritis patients' inflammation.
The rows are the individual patients and the columns
are their daily inflammation measurements.

265
> ## Data Type
266
267
268
269
>
> A Numpy array contains one or more elements
> of the same type. `type` will only tell you that
> a variable is a NumPy array.
Valentina Staneva's avatar
Valentina Staneva committed
270
> We can also find out the type
271
272
> of the data contained in the NumPy array.
>
273
> ~~~
274
275
> print(data.dtype)
> ~~~
276
277
278
> {: .python}
>
> ~~~
279
280
> dtype('float64')
> ~~~
281
> {: .output}
282
283
>
> This tells us that the NumPy array's elements are
284
> [floating-point numbers]({{ page.root }}/reference/#floating-point number).
285
{: .callout}
286

287
We can see what the array's [shape]({{ page.root }}/reference/#shape) is like this:
288

289
~~~
290
print(data.shape)
Greg Wilson's avatar
Greg Wilson committed
291
~~~
292
293
294
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
295
296
(60, 40)
~~~
297
{: .output}
298

299
300
This tells us that `data` has 60 rows and 40 columns. When we created the
variable `data` to store our arthritis data, we didn't just create the array, we also
301
created information about the array, called [members]({{ page.root }}/reference/#member) or
302
303
attributes. This extra information describes `data` in
the same way an adjective describes a noun.
Bartosz T's avatar
Bartosz T committed
304
`data.shape` is an attribute  of `data` which describes the dimensions of `data`.
305
We use the same dotted notation for the attributes of variables
306
307
that we use for the functions in libraries
because they have the same part-and-whole relationship.
308

309
If we want to get a single number from the array,
310
we must provide an [index]({{ page.root }}/reference/#index) in square brackets,
311
just as we do in math:
312

313
~~~
314
print('first value in data:', data[0, 0])
Greg Wilson's avatar
Greg Wilson committed
315
~~~
316
317
318
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
319
320
first value in data: 0.0
~~~
321
{: .output}
322

323
~~~
324
print('middle value in data:', data[30, 20])
Greg Wilson's avatar
Greg Wilson committed
325
~~~
326
327
328
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
329
330
middle value in data: 13.0
~~~
331
{: .output}
332

333
334
The expression `data[30, 20]` may not surprise you,
but `data[0, 0]` might.
335
Programming languages like Fortran, MATLAB and R start counting at 1,
336
337
because that's what human beings have done for thousands of years.
Languages in the C family (including C++, Java, Perl, and Python) count from 0
338
339
340
341
342
because it represents an offset from the first value in the array (the second
value is offset by one index from the first value). This is closer to the way
that computers represent arrays (if you are interested in the historical
reasons behind counting indices from zero, you can read
[Mike Hoye's blog post](http://exple.tive.org/blarg/2013/10/22/citation-needed/)).
343
As a result,
Greg Wilson's avatar
Greg Wilson committed
344
if we have an M×N array in Python,
345
346
347
348
349
350
its indices go from 0 to M-1 on the first axis
and 0 to N-1 on the second.
It takes a bit of getting used to,
but one way to remember the rule is that
the index is how many steps we have to take from the start to get the item we want.

351
> ## In the Corner
352
353
354
355
356
357
>
> What may also surprise you is that when Python displays an array,
> it shows the element with index `[0, 0]` in the upper left corner
> rather than the lower left.
> This is consistent with the way mathematicians draw matrices,
> but different from the Cartesian coordinates.
358
> The indices are (row, column) instead of (column, row) for the same reason,
359
> which can be confusing when plotting data.
360
{: .callout}
361
362
363
364
365

An index like `[30, 20]` selects a single element of an array,
but we can select whole sections as well.
For example,
we can select the first ten days (columns) of values
366
for the first four patients (rows) like this:
367

368
~~~
369
print(data[0:4, 0:10])
Greg Wilson's avatar
Greg Wilson committed
370
~~~
371
372
373
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
374
[[ 0.  0.  1.  3.  1.  2.  4.  7.  8.  3.]
375
376
377
 [ 0.  1.  2.  1.  2.  1.  3.  2.  2.  6.]
 [ 0.  1.  1.  3.  3.  2.  6.  2.  5.  9.]
 [ 0.  0.  2.  0.  4.  2.  2.  1.  6.  7.]]
Greg Wilson's avatar
Greg Wilson committed
378
~~~
379
{: .output}
380

381
The [slice]({{ page.root }}/reference/#slice) `0:4` means,
382
383
384
385
386
387
"Start at index 0 and go up to, but not including, index 4."
Again,
the up-to-but-not-including takes a bit of getting used to,
but the rule is that the difference between the upper and lower bounds is the number of values in the slice.

We don't have to start slices at 0:
388

389
~~~
390
print(data[5:10, 0:10])
Greg Wilson's avatar
Greg Wilson committed
391
~~~
392
393
394
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
395
[[ 0.  0.  1.  2.  2.  4.  2.  1.  6.  4.]
396
397
398
399
 [ 0.  0.  2.  2.  4.  2.  2.  5.  5.  8.]
 [ 0.  0.  1.  2.  3.  1.  2.  3.  5.  3.]
 [ 0.  0.  0.  3.  1.  5.  6.  5.  5.  8.]
 [ 0.  1.  1.  2.  1.  3.  5.  3.  5.  8.]]
Greg Wilson's avatar
Greg Wilson committed
400
~~~
401
{: .output}
402

403
404
405
406
407
408
409
410
We also don't have to include the upper and lower bound on the slice.
If we don't include the lower bound,
Python uses 0 by default;
if we don't include the upper,
the slice runs to the end of the axis,
and if we don't include either
(i.e., if we just use ':' on its own),
the slice includes everything:
411

412
~~~
Greg Wilson's avatar
Greg Wilson committed
413
small = data[:3, 36:]
414
415
print('small is:')
print(small)
Greg Wilson's avatar
Greg Wilson committed
416
~~~
417
418
419
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
420
small is:
421
422
423
[[ 2.  3.  0.  0.]
 [ 1.  1.  0.  1.]
 [ 2.  2.  1.  1.]]
Greg Wilson's avatar
Greg Wilson committed
424
~~~
425
{: .output}
426

427
Arrays also know how to perform common mathematical operations on their values.
Greg Wilson's avatar
Greg Wilson committed
428
429
430
431
432
The simplest operations with data are arithmetic:
add, subtract, multiply, and divide.
 When you do such operations on arrays,
the operation is done on each individual element of the array.
Thus:
433

434
~~~
Greg Wilson's avatar
Greg Wilson committed
435
436
doubledata = data * 2.0
~~~
437
{: .python}
438

Greg Wilson's avatar
Greg Wilson committed
439
440
will create a new array `doubledata`
whose elements have the value of two times the value of the corresponding elements in `data`:
441

442
~~~
443
444
445
446
print('original:')
print(data[:3, 36:])
print('doubledata:')
print(doubledata[:3, 36:])
Greg Wilson's avatar
Greg Wilson committed
447
~~~
448
449
450
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
451
original:
452
453
454
455
456
457
458
[[ 2.  3.  0.  0.]
 [ 1.  1.  0.  1.]
 [ 2.  2.  1.  1.]]
doubledata:
[[ 4.  6.  0.  0.]
 [ 2.  2.  0.  2.]
 [ 4.  4.  2.  2.]]
Greg Wilson's avatar
Greg Wilson committed
459
~~~
460
{: .output}
461

Greg Wilson's avatar
Greg Wilson committed
462
463
If,
instead of taking an array and doing arithmetic with a single value (as above)
464
you did the arithmetic operation with another array of the same shape,
Greg Wilson's avatar
Greg Wilson committed
465
466
the operation will be done on corresponding elements of the two arrays.
Thus:
467

468
~~~
Greg Wilson's avatar
Greg Wilson committed
469
470
tripledata = doubledata + data
~~~
471
{: .python}
472

473
474
475
will give you an array where `tripledata[0,0]` will equal `doubledata[0,0]` plus `data[0,0]`,
and so on for all other elements of the arrays.

476
~~~
477
478
print('tripledata:')
print(tripledata[:3, 36:])
Greg Wilson's avatar
Greg Wilson committed
479
~~~
480
481
482
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
483
tripledata:
484
485
486
[[ 6.  9.  0.  0.]
 [ 3.  3.  0.  3.]
 [ 6.  6.  3.  3.]]
Greg Wilson's avatar
Greg Wilson committed
487
~~~
488
{: .output}
489

490
Often, we want to do more than add, subtract, multiply, and divide values of data.
491
NumPy knows how to do more complex operations on arrays.
492
493
If we want to find the average inflammation for all patients on all days,
for example,
494
we can ask NumPy to compute `data`'s mean value:
495

496
~~~
497
print(numpy.mean(data))
Greg Wilson's avatar
Greg Wilson committed
498
~~~
499
500
501
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
502
503
6.14875
~~~
504
{: .output}
505

506
507
`mean` is a [function]({{ page.root }}/reference/#function) that takes
an array as an [argument]({{ page.root }}/reference/#argument).
508
509
510
If variables are nouns, functions are verbs:
they do things with variables.

511
> ## Not All Functions Have Input
512
513
514
>
> Generally, a function uses inputs to produce outputs.
> However, some functions produce outputs without
515
516
> needing any input. For example, checking the current time
> doesn't require any input.
517
>
518
> ~~~
519
520
> import time
> print(time.ctime())
521
> ~~~
522
523
524
> {: .python}
>
> ~~~
525
> 'Sat Mar 26 13:07:33 2016'
526
> ~~~
527
> {: .output}
528
529
530
531
>
> For functions that don't take in any arguments,
> we still need parentheses (`()`)
> to tell Python to go and do something for us.
532
{: .callout}
533
534
535

NumPy has lots of useful functions that take an array as input.
Let's use three of those functions to get some descriptive values about the dataset.
536
537
We'll also use multiple assignment,
a convenient Python feature that will enable us to do this all in one line.
538

539
~~~
540
maxval, minval, stdval = numpy.max(data), numpy.min(data), numpy.std(data)
541

Alistair Walsh's avatar
Alistair Walsh committed
542
543
544
print('maximum inflammation:', maxval)
print('minimum inflammation:', minval)
print('standard deviation:', stdval)
Greg Wilson's avatar
Greg Wilson committed
545
~~~
546
547
548
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
549
maximum inflammation: 20.0
550
551
minimum inflammation: 0.0
standard deviation: 4.61383319712
Greg Wilson's avatar
Greg Wilson committed
552
~~~
553
{: .output}
554

555
> ## Mystery Functions in IPython
556
>
557
558
559
560
561
562
563
> How did we know what functions NumPy has and how to use them?
> If you are working in the IPython/Jupyter Notebook there is an easy way to find out.
> If you type the name of something with a full-stop then you can use tab completion
> (e.g. type `numpy.` and then press tab)
> to see a list of all functions and attributes that you can use. After selecting one you
> can also add a question mark (e.g. `numpy.cumprod?`) and IPython will return an
> explanation of the method! This is the same as doing `help(numpy.cumprod)`.
564
{: .callout}
565
566

When analyzing data, though,
567
568
569
we often want to look at partial statistics,
such as the maximum value per patient
or the average value per day.
570
One way to do this is to create a new temporary array of the data we want,
571
then ask it to do the calculation:
572

573
~~~
Greg Wilson's avatar
Greg Wilson committed
574
patient_0 = data[0, :] # 0 on the first axis, everything on the second
575
print('maximum inflammation for patient 0:', patient_0.max())
Greg Wilson's avatar
Greg Wilson committed
576
~~~
577
578
579
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
580
581
maximum inflammation for patient 0: 18.0
~~~
582
{: .output}
583

584
Everything in a line of code following the '#' symbol is a
585
[comment]({{ page.root }}/reference/#comment) that is ignored by the computer.
586
Comments allow programmers to leave explanatory notes for other
jstapleton's avatar
jstapleton committed
587
588
programmers or their future selves.

589
We don't actually need to store the row in a variable of its own.
590
Instead, we can combine the selection and the function call:
591

592
~~~
593
print('maximum inflammation for patient 2:', numpy.max(data[2, :]))
Greg Wilson's avatar
Greg Wilson committed
594
~~~
595
596
597
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
598
599
maximum inflammation for patient 2: 19.0
~~~
600
{: .output}
601

602
What if we need the maximum inflammation for each patient over all days (as in the
603
604
605
next diagram on the left), or the average for each day (as in the
diagram on the right)? As the diagram below shows, we want to perform the
operation across an axis:
606

607
![Operations Across Axes](../fig/python-operations-across-axes.png)
608

609
To support this,
610
most array functions allow us to specify the axis we want to work on.
611
If we ask for the average across axis 0 (rows in our 2D example),
612
we get:
613

614
~~~
615
print(numpy.mean(data, axis=0))
Greg Wilson's avatar
Greg Wilson committed
616
~~~
617
618
619
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
620
[  0.           0.45         1.11666667   1.75         2.43333333   3.15
621
622
623
624
625
626
627
   3.8          3.88333333   5.23333333   5.51666667   5.95         5.9
   8.35         7.73333333   8.36666667   9.5          9.58333333
  10.63333333  11.56666667  12.35        13.25        11.96666667
  11.03333333  10.16666667  10.           8.66666667   9.15         7.25
   7.33333333   6.58333333   6.06666667   5.95         5.11666667   3.6
   3.3          3.56666667   2.48333333   1.5          1.13333333
   0.56666667]
Greg Wilson's avatar
Greg Wilson committed
628
~~~
629
{: .output}
630

631
632
As a quick check,
we can ask this array what its shape is:
633

634
~~~
635
print(numpy.mean(data, axis=0).shape)
Greg Wilson's avatar
Greg Wilson committed
636
~~~
637
638
639
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
640
641
(40,)
~~~
642
{: .output}
643

Greg Wilson's avatar
Greg Wilson committed
644
The expression `(40,)` tells us we have an N×1 vector,
645
so this is the average inflammation per day for all patients.
646
If we average across axis 1 (columns in our 2D example), we get:
647

648
~~~
649
print(numpy.mean(data, axis=1))
Greg Wilson's avatar
Greg Wilson committed
650
~~~
651
652
653
{: .python}

~~~
Greg Wilson's avatar
Greg Wilson committed
654
[ 5.45   5.425  6.1    5.9    5.55   6.225  5.975  6.65   6.625  6.525
655
656
657
658
659
  6.775  5.8    6.225  5.75   5.225  6.3    6.55   5.7    5.85   6.55
  5.775  5.825  6.175  6.1    5.8    6.425  6.05   6.025  6.175  6.55
  6.175  6.35   6.725  6.125  7.075  5.725  5.925  6.15   6.075  5.75
  5.975  5.725  6.3    5.9    6.75   5.925  7.225  6.15   5.95   6.275  5.7
  6.1    6.825  5.975  6.725  5.7    6.25   6.4    7.05   5.9  ]
Greg Wilson's avatar
Greg Wilson committed
660
~~~
661
{: .output}
662

663
664
665
666
667
which is the average inflammation per patient across all days.

The mathematician Richard Hamming once said,
"The purpose of computing is insight, not numbers,"
and the best way to develop insight is often to visualize data.
Valentina Staneva's avatar
Valentina Staneva committed
668
Visualization deserves an entire lecture (of course) of its own,
669
but we can explore a few features of Python's `matplotlib` library here.
Greg Wilson's avatar
Greg Wilson committed
670
671
While there is no "official" plotting library,
this package is the de facto standard.
672
673
674
First,
we will import the `pyplot` module from `matplotlib`
and use two of its functions to create and display a heat map of our data:
675

676
~~~
677
import matplotlib.pyplot
678
image = matplotlib.pyplot.imshow(data)
679
matplotlib.pyplot.show()
Greg Wilson's avatar
Greg Wilson committed
680
~~~
681
{: .python}
682

683
![Heatmap of the Data](../fig/01-numpy_71_0.png)
684

685
686
687
Blue regions in this heat map are low values, while red shows high values.
As we can see,
inflammation rises and falls over a 40-day period.
688

689
> ## Some IPython Magic
690
691
692
693
>
> If you're using an IPython / Jupyter notebook,
> you'll need to execute the following command
> in order for your matplotlib images to appear
Damien Irving's avatar
Damien Irving committed
694
> in the notebook when `show()` is called:
Damien Irving's avatar
Damien Irving committed
695
>
696
> ~~~
697
> %matplotlib inline
Damien Irving's avatar
Damien Irving committed
698
> ~~~
699
> {: .python}
700
>
701
702
> The `%` indicates an IPython magic function -
> a function that is only valid within the notebook environment.
703
> Note that you only have to execute this function once per notebook.
704
{: .callout}
705

706
Let's take a look at the average inflammation over time:
707

708
~~~
709
ave_inflammation = numpy.mean(data, axis=0)
710
ave_plot = matplotlib.pyplot.plot(ave_inflammation)
711
matplotlib.pyplot.show()
Greg Wilson's avatar
Greg Wilson committed
712
~~~
713
{: .python}
714

715
![Average Inflammation Over Time](../fig/01-numpy_73_0.png)
716

717
718
Here,
we have put the average per day across all patients in the variable `ave_inflammation`,
719
then asked `matplotlib.pyplot` to create and display a line graph of those values.
720
721
722
723
724
The result is roughly a linear rise and fall,
which is suspicious:
based on other studies,
we expect a sharper rise and slower fall.
Let's have a look at two other statistics:
725

726
~~~
727
max_plot = matplotlib.pyplot.plot(numpy.max(data, axis=0))
728
matplotlib.pyplot.show()
Greg Wilson's avatar
Greg Wilson committed
729
~~~
730
{: .python}
731

732
![Maximum Value Along The First Axis](../fig/01-numpy_75_1.png)
Greg Wilson's avatar
Greg Wilson committed
733

734
~~~
735
min_plot = matplotlib.pyplot.plot(numpy.min(data, axis=0))
736
matplotlib.pyplot.show()
Greg Wilson's avatar
Greg Wilson committed
737
~~~
738
{: .python}
739

740
![Minimum Value Along The First Axis](../fig/01-numpy_75_3.png)
741

742
743
744
745
746
The maximum value rises and falls perfectly smoothly,
while the minimum seems to be a step function.
Neither result seems particularly likely,
so either there's a mistake in our calculations
or something is wrong with our data.
747
This insight would have been difficult to reach by
jstapleton's avatar
jstapleton committed
748
examining the data without visualization tools.
749

750
You can group similar plots in a single figure using subplots.
751
This script below uses a number of new commands. The function `matplotlib.pyplot.figure()`
752
creates a space into which we will place all of our plots. The parameter `figsize`
753
tells Python how big to make this space. Each subplot is placed into the figure using
754
its `add_subplot` [method]({{ page.root }}/reference/#method). The `add_subplot` method takes 3 parameters. The first denotes
755
how many total rows of subplots there are, the second parameter refers to the
valiseverywhere's avatar
valiseverywhere committed
756
total number of subplot columns, and the final parameter denotes which subplot
757
758
759
your variable is referencing (left-to-right, top-to-bottom). Each subplot is stored in a
different variable (`axes1`, `axes2`, `axes3`). Once a subplot is created, the axes can
be titled using the `set_xlabel()` command (or `set_ylabel()`).
760
Here are our three plots side by side:
761

762
~~~
763
764
import numpy
import matplotlib.pyplot
765

766
data = numpy.loadtxt(fname='inflammation-01.csv', delimiter=',')
767

768
fig = matplotlib.pyplot.figure(figsize=(10.0, 3.0))
769

770
771
772
axes1 = fig.add_subplot(1, 3, 1)
axes2 = fig.add_subplot(1, 3, 2)
axes3 = fig.add_subplot(1, 3, 3)
773

774
axes1.set_ylabel('average')
775
axes1.plot(numpy.mean(data, axis=0))
776

777
axes2.set_ylabel('max')
778
axes2.plot(numpy.max(data, axis=0))
779

780
axes3.set_ylabel('min')
781
axes3.plot(numpy.min(data, axis=0))
782
783

fig.tight_layout()
784

785
matplotlib.pyplot.show()
Greg Wilson's avatar
Greg Wilson committed
786
~~~
787
{: .python}
788

789
![The Previous Plots as Subplots](../fig/01-numpy_80_0.png)
790

791
The [call]({{ page.root }}/reference/#function-call) to `loadtxt` reads our data,
792
793
and the rest of the program tells the plotting library
how large we want the figure to be,
794
that we're creating three subplots,
795
796
797
what to draw for each one,
and that we want a tight layout.
(Perversely,
798
if we leave out that call to `fig.tight_layout()`,
799
the graphs will actually be squeezed together more closely.)
800

801
> ## Scientists Dislike Typing
802
803
804
805
806
807
808
>
> We will always use the syntax `import numpy` to import NumPy.
> However, in order to save typing, it is
> [often suggested](http://www.scipy.org/getting-started.html#an-example-script)
> to make a shortcut like so: `import numpy as np`.
> If you ever see Python code online using a NumPy function with `np`
> (for example, `np.loadtxt(...)`), it's because they've used this shortcut.
809
> When working with other people, it is important to agree on a convention of how common libraries are imported.
valiseverywhere's avatar
valiseverywhere committed
810
{: .callout}
811
> ## Check Your Understanding
mboisson's avatar
mboisson committed
812
813
814
>
> Draw diagrams showing what variables refer to what values after each statement in the following program:
>
815
> ~~~
mboisson's avatar
mboisson committed
816
817
818
819
820
> mass = 47.5
> age = 122
> mass = mass * 2.0
> age = age - 20
> ~~~
821
822
> {: .python}
{: .challenge}
mboisson's avatar
mboisson committed
823

824
> ## Sorting Out References
mboisson's avatar
mboisson committed
825
826
827
>
> What does the following program print out?
>
828
> ~~~
mboisson's avatar
mboisson committed
829
830
> first, second = 'Grace', 'Hopper'
> third, fourth = second, first
831
> print(third, fourth)
mboisson's avatar
mboisson committed
832
> ~~~
833
> {: .python}
834
835
836
837
838
839
840
>
> > ## Solution
> > ~~~
> > Hopper Grace
> > ~~~
> > {: .output}
> {: .solution}
841
{: .challenge}
mboisson's avatar
mboisson committed
842

843
> ## Slicing Strings
mboisson's avatar
mboisson committed
844
>