4.3: Vectors in Python
- Page ID
- 63857
For those who are new to Python, there are many common mistakes happen in this course. Try to fix the following codes.
SyntaxError
It means that the code does not make sense in Python. We would like to define a vector with four numbers.
Fix the following code to creat three vectors with four numbers.
Although you may have been able to get rid of the error messages the answer to you problem may still not be correct. Throughout the semester we will be using a python program called answercheck
to allow you to check some of your answers. This program doesn’t tell you the right answer but it is intended to be used as a way to get immediate feedback and accelerate learning.
First we will need to download answercheck.py
to your current working directory. You only really need to do this once. However, if you delete this file by mistake sometime during the semester, you can come back to this notebook and download it again by running the following cell:
How just run the following command to see if you got \(x\), \(y\) and \(z\) correct when you fixed the code above.
make sure you do not change the checkanswer
commands. The long string with numbers and letters is the secret code that encodes the true answer. This code is also called the HASH. Feel free to look at the answercheck.py
code and see if you can figure out how it works?
Numpy
Numpy is a common way to represent vectors, and you are suggested to use numpy
unless otherwise specified. The benefit of numpy
is that it can perform the linear algebra operations listed in the previous section.
For example, the following code uses numpy.array
to define a vector of four elements.
Scalars versus 1-vectors
In mathematics, 1-vector is considered as a scalar. But in Python, they are not the same.
Lists of vectors
We have a list of numpy arrays or a list of list. In this case, the vectors can have different dimensions.
Modify the print statement using indexing to only print the value 3 from the list_of_vectors
defined below.
Indexing
The index of a vector runs from 0 to \(n−1\) for a \(n\)-vector.
The following code tries to get the third element of x_np
, which is the number 2.0
. Fix this code to provide the correct answer.
Replace only the third element of x_np
with the number 20.0
such that the new values of x_np
is [-1, 0, 20., 3.1]
There is a special index -1, which represents the last element in an array. There are several ways to get more than one consecutive elements.
- x_np[1:3] gives the 2nd and 3rd elements only. It starts with the first index and ends before the second index. So the number of element is just the difference between these two numbers.
- x_np[1:-1] is the same as x_np[1:3] for a 4-vector.
- If you want the last element also, then you do not need to put the index, e.g., x_n[1:] gives all elements except the first one. You can do the same thing as the first one.
you are given a vector (x_np
) of \(n\) elements, define a new vector (d
) of size \(n−1\) such that \( d_i = x_{i+1} - x_i \) for \( i=1, \dots, n-1 \).
Hint try doing this without writing your own loop. You should be able to use simple numpy
indexing as described above.
Assignment versus copying
Take a look at the following code.
- we create one numpy array
x_np
- we let
y_np = x_np
- we change the third element of
y_np
- The third element of
x_np
is also changed
This looks weired and may not make sense for those uses other languages such as MATLAB.
The reason for this is that we are not creating a copy of x_np
and name it as y_np
. What we did is that we give a new name y_np
to the same array x_np
. Therefore, if one is changed, and the other one is also changed, because they refer to the same array.
There is a method named copy
that can be used to create a new array. You can search how it works and fix the code below. If this is done correctly the x_np
vector should stay the same and the y_np
you now be [-1 0 2 3.1]
.
Vector equality in numpy and list
The relational operator (==
, <
, >
, !=
, etc.) can be used to check whether the vectors are same or not. However, they will act differently if the code is comparing numpy.array
objects or a list
. In numpy
, In numpy
relational operators checks the equality for each element in the array
. For list
, relational operators check all elements.
Zero vectors and Ones vectors in numpy
- zeros(n) creates a vector with all 0s
- ones(n) creates a vector with all 1s
Create a zero vector (called zero_np
) with the same dimension as vector x_np
. Create a ones vector (called ones+np
) also with the same dimension as vector x_np
.
Random vectors
- random.random(n) creates a random vector with dimension \(n\).
Vector addition and subtraction
In this section, you will understand why we use numpy for linear algebra opeartions. If x and y are numpy arrays of the same size, we can have x + y and x-y for their addition and subtraction, respectively.
For comparison, we also put the addition of two lists below. Recall from the pre-class assignment, we have to define a function to add two lists for linear algebra.
Modify the following code to properly add and subtract the two lists.
HINT it is perfectly okay NOT to write your own function try you should be able to cast the lists as arrays:
Scalar-vector addition
A scalar-vector addition means that the scalar (or a 1-vector) is added to all elements of the vector.
Add a scalar 20.20 to all elements of the following vector x_np
and store teh result back into x_np
Scalar-vector multiplication and division
When a
is a scalar and x
is numpy
array. We can express the scalar-vector multiplication as a*x
or x*a
.
We can also do scalar-vector division for x/a
or a/x
. (note that x/a
and a/x
are different)
Divide all elements of the following vector x_np
by 20.20
and put it into y_np
Element-wise operations
As stated above relational operations on numpy
arrays are performed element-wise. Examples we mentioned before are
- The
==
operator - The addition
+
and subtraction-
for this to work the two vectors have to be the same dimensions.
If they are not have the same dimension, such as a scalar and a vector, we can think about expanding the scalar to have the same dimension as the vector and perform the operations. For example.
- Vector-scalar addition and subtraction
- Vector-scalar multiplication and division
Assume that you invested three assets with initial values stored in p_initial
, and after one week, their values are stored in p_final
. Then what are the asset return ratio (r
) for these three assets (i.e. price change over the initial value).
Linear combination
We have two vectors \(x\) and \(y\) we can get the linear combination of these two vectors as \(ax+by\) where \(a\) and \(b\) are scalar coefficients.
In the following example, we are given two vectors (x_np
and y_np
), and two scalars (alpha
and beta
), we obtain the linear combination alpha*x_np + beta*y_np
.
We can also define a function lincomb
to performn the linear combination.
Finish the following code for lincomb and compare the results we just get.
We can also test the functions ourselves by using values for which we know the answer. For example, the following tests are multiplying and adding by zero we know what these answers should be and can check them.
If you want to check that all values in a numpy.array
are the same you could convert it to a list or there is a method called alltrue
which checks if everything is true. It is a good idea to use this method if vectors get big.