Search

Text Color

Margin Size

Font Type

Enable Dyslexic Font

4.3: Vectors in Python

Last updated

Sep 17, 2022
Save as PDF
- 4.2: Pre-class assignment review
- 5: 03 Pre-Class Assignment - Linear Equations

$\newcommand{\vecs}[1]{\overset { \scriptstyle \rightharpoonup} {\mathbf{#1}} }$

$\newcommand{\vecd}[1]{\overset{-\!-\!\rightharpoonup}{\vphantom{a}\smash {#1}}}$

$\newcommand{\id}{\mathrm{id}}$ $\newcommand{\Span}{\mathrm{span}}$

( \newcommand{\kernel}{\mathrm{null}\,}\) $\newcommand{\range}{\mathrm{range}\,}$

$\newcommand{\RealPart}{\mathrm{Re}}$ $\newcommand{\ImaginaryPart}{\mathrm{Im}}$

$\newcommand{\Argument}{\mathrm{Arg}}$ $\newcommand{\norm}[1]{\| #1 \|}$

$\newcommand{\inner}[2]{\langle #1, #2 \rangle}$

$\newcommand{\Span}{\mathrm{span}}$

$\newcommand{\id}{\mathrm{id}}$

$\newcommand{\Span}{\mathrm{span}}$

$\newcommand{\kernel}{\mathrm{null}\,}$

$\newcommand{\range}{\mathrm{range}\,}$

$\newcommand{\RealPart}{\mathrm{Re}}$

$\newcommand{\ImaginaryPart}{\mathrm{Im}}$

$\newcommand{\Argument}{\mathrm{Arg}}$

$\newcommand{\norm}[1]{\| #1 \|}$

$\newcommand{\inner}[2]{\langle #1, #2 \rangle}$

$\newcommand{\Span}{\mathrm{span}}$ $\newcommand{\AA}{\unicode[.8,0]{x212B}}$

$\newcommand{\vectorA}[1]{\vec{#1}} % arrow$

$\newcommand{\vectorAt}[1]{\vec{\text{#1}}} % arrow$

$\newcommand{\vectorB}[1]{\overset { \scriptstyle \rightharpoonup} {\mathbf{#1}} }$

$\newcommand{\vectorC}[1]{\textbf{#1}}$

$\newcommand{\vectorD}[1]{\overrightarrow{#1}}$

$\newcommand{\vectorDt}[1]{\overrightarrow{\text{#1}}}$

$\newcommand{\vectE}[1]{\overset{-\!-\!\rightharpoonup}{\vphantom{a}\smash{\mathbf {#1}}}}$

$\newcommand{\vecs}[1]{\overset { \scriptstyle \rightharpoonup} {\mathbf{#1}} }$

$\newcommand{\vecd}[1]{\overset{-\!-\!\rightharpoonup}{\vphantom{a}\smash {#1}}}$

$\newcommand{\avec}{\mathbf a}$

$\newcommand{\bvec}{\mathbf b}$

$\newcommand{\cvec}{\mathbf c}$

$\newcommand{\dvec}{\mathbf d}$

$\newcommand{\dtil}{\widetilde{\mathbf d}}$

$\newcommand{\evec}{\mathbf e}$

$\newcommand{\fvec}{\mathbf f}$

$\newcommand{\nvec}{\mathbf n}$

$\newcommand{\pvec}{\mathbf p}$

$\newcommand{\qvec}{\mathbf q}$

$\newcommand{\svec}{\mathbf s}$

$\newcommand{\tvec}{\mathbf t}$

$\newcommand{\uvec}{\mathbf u}$

$\newcommand{\vvec}{\mathbf v}$

$\newcommand{\wvec}{\mathbf w}$

$\newcommand{\xvec}{\mathbf x}$

$\newcommand{\yvec}{\mathbf y}$

$\newcommand{\zvec}{\mathbf z}$

$\newcommand{\rvec}{\mathbf r}$

$\newcommand{\mvec}{\mathbf m}$

$\newcommand{\zerovec}{\mathbf 0}$

$\newcommand{\onevec}{\mathbf 1}$

$\newcommand{\real}{\mathbb R}$

$\newcommand{\twovec}[2]{\left[\begin{array}{r}#1 \\ #2 \end{array}\right]}$

$\newcommand{\ctwovec}[2]{\left[\begin{array}{c}#1 \\ #2 \end{array}\right]}$

$\newcommand{\threevec}[3]{\left[\begin{array}{r}#1 \\ #2 \\ #3 \end{array}\right]}$

$\newcommand{\cthreevec}[3]{\left[\begin{array}{c}#1 \\ #2 \\ #3 \end{array}\right]}$

$\newcommand{\fourvec}[4]{\left[\begin{array}{r}#1 \\ #2 \\ #3 \\ #4 \end{array}\right]}$

$\newcommand{\cfourvec}[4]{\left[\begin{array}{c}#1 \\ #2 \\ #3 \\ #4 \end{array}\right]}$

$\newcommand{\fivevec}[5]{\left[\begin{array}{r}#1 \\ #2 \\ #3 \\ #4 \\ #5 \\ \end{array}\right]}$

$\newcommand{\cfivevec}[5]{\left[\begin{array}{c}#1 \\ #2 \\ #3 \\ #4 \\ #5 \\ \end{array}\right]}$

$\newcommand{\mattwo}[4]{\left[\begin{array}{rr}#1 \amp #2 \\ #3 \amp #4 \\ \end{array}\right]}$

$\newcommand{\laspan}[1]{\text{Span}\{#1\}}$

$\newcommand{\bcal}{\cal B}$

$\newcommand{\ccal}{\cal C}$

$\newcommand{\scal}{\cal S}$

$\newcommand{\wcal}{\cal W}$

$\newcommand{\ecal}{\cal E}$

$\newcommand{\coords}[2]{\left\{#1\right\}_{#2}}$

$\newcommand{\gray}[1]{\color{gray}{#1}}$

$\newcommand{\lgray}[1]{\color{lightgray}{#1}}$

$\newcommand{\rank}{\operatorname{rank}}$

$\newcommand{\row}{\text{Row}}$

$\newcommand{\col}{\text{Col}}$

$\renewcommand{\row}{\text{Row}}$

$\newcommand{\nul}{\text{Nul}}$

$\newcommand{\var}{\text{Var}}$

$\newcommand{\corr}{\text{corr}}$

$\newcommand{\len}[1]{\left|#1\right|}$

$\newcommand{\bbar}{\overline{\bvec}}$

$\newcommand{\bhat}{\widehat{\bvec}}$

$\newcommand{\bperp}{\bvec^\perp}$

$\newcommand{\xhat}{\widehat{\xvec}}$

$\newcommand{\vhat}{\widehat{\vvec}}$

$\newcommand{\uhat}{\widehat{\uvec}}$

$\newcommand{\what}{\widehat{\wvec}}$

$\newcommand{\Sighat}{\widehat{\Sigma}}$

$\newcommand{\lt}{<}$

$\newcommand{\gt}{>}$

$\newcommand{\amp}{&}$

$\definecolor{fillinmathshade}{gray}{0.9}$

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.

Do This

Fix the following code to creat three vectors with four numbers.

If you have already signed in, please refresh the page.

x = [1 2 3.4 4]
y = [1, 2, 3, 5]]
z = [[1, 2, 3, 6.3]

x = [1 2 3.4 4]
y = [1, 2, 3, 5]]
z = [[1, 2, 3, 6.3]

  File "<ipython-input-1-cd07b469c255>", line 1
    x = [1 2 3.4 4]
           ^
SyntaxError: invalid syntax

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.

Do This

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:

If you have already signed in, please refresh the page.

from urllib.request import urlretrieve

urlretrieve('https://raw.githubusercontent.com/colbrydi/jupytercheck/master/answercheck.py', 
            'answercheck.py');

from urllib.request import urlretrieve

urlretrieve('https://raw.githubusercontent.com/colbrydi/jupytercheck/master/answercheck.py', 
            'answercheck.py');

Do This

How just run the following command to see if you got $x$ , $y$ and $z$ correct when you fixed the code above.

If you have already signed in, please refresh the page.

from answercheck import checkanswer

checkanswer([x,y,z],'e80321644979a873b273aebbbcd0e450');

from answercheck import checkanswer

checkanswer([x,y,z],'e80321644979a873b273aebbbcd0e450');

Note

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.

If you have already signed in, please refresh the page.

import numpy as np
x_np = np.array([-1, 0, 2, 3.1])
x_np

import numpy as np
x_np = np.array([-1, 0, 2, 3.1])
x_np

Scalars versus 1-vectors

In mathematics, 1-vector is considered as a scalar. But in Python, they are not the same.

If you have already signed in, please refresh the page.

x = 2.4
y = [2.4]
x == y

x = 2.4
y = [2.4]
x == y

False

If you have already signed in, please refresh the page.

x == y[0]

x == y[0]

True

Lists of vectors

We have a list of numpy arrays or a list of list. In this case, the vectors can have different dimensions.

Do This

Modify the print statement using indexing to only print the value 3 from the list_of_vectors defined below.

If you have already signed in, please refresh the page.

x_np = np.array([-1,0, 2 , 3.1])
y_np = np.array([1,-1,3])
z_np = np.array([0,1])
list_of_vectors = [x_np,y_np,z_np]

print(list_of_vectors)

x_np = np.array([-1,0, 2 , 3.1])
y_np = np.array([1,-1,3])
z_np = np.array([0,1])
list_of_vectors = [x_np,y_np,z_np]

print(list_of_vectors)

[array([-1. ,  0. ,  2. ,  3.1]), array([ 1, -1,  3]), array([0, 1])]

Indexing

The index of a vector runs from 0 to $n−1$ for a $n$ -vector.

Do This

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.

If you have already signed in, please refresh the page.

print(x_np(3))

print(x_np(3))

Do This

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]

If you have already signed in, please refresh the page.

# Replace the third element using 20.0, then the resulting element is 
#####Start your code here #####

#####End of your code here#####
print(x_np)

# Replace the third element using 20.0, then the resulting element is 
#####Start your code here #####

#####End of your code here#####
print(x_np)

If you have already signed in, please refresh the page.

from answercheck import checkanswer

checkanswer(x_np,'993d5cbc6ddeb10776ed48159780a5d3');

from answercheck import checkanswer

checkanswer(x_np,'993d5cbc6ddeb10776ed48159780a5d3');

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.

Do This

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.

If you have already signed in, please refresh the page.

x_np = np.array([1,8,3,2,1,9,7])

## Put your answer to the above question here.

x_np = np.array([1,8,3,2,1,9,7])

## Put your answer to the above question here.

If you have already signed in, please refresh the page.

from answercheck import checkanswer

checkanswer(d,'14205415f0ed56e608d0a87e7253fa70');

from answercheck import checkanswer

checkanswer(d,'14205415f0ed56e608d0a87e7253fa70');

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.

If you have already signed in, please refresh the page.

x_np = np.array([-1, 0, 2, 3.1])
y_np = x_np
y_np[2] = 20.0
x_np

x_np = np.array([-1, 0, 2, 3.1])
y_np = x_np
y_np[2] = 20.0
x_np

Do This

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].

If you have already signed in, please refresh the page.

## modify the following code to copy the x_np instead of just giving it a new name
x_np = np.array([-1, 0, 2, 3.1])
y_np = x_np
y_np[2] = 20.0
print(x_np)

## modify the following code to copy the x_np instead of just giving it a new name
x_np = np.array([-1, 0, 2, 3.1])
y_np = x_np
y_np[2] = 20.0
print(x_np)

If you have already signed in, please refresh the page.

from answercheck import checkanswer

checkanswer(x_np,'0ba269d18692155ba252e6addedf37ad');

from answercheck import checkanswer

checkanswer(x_np,'0ba269d18692155ba252e6addedf37ad');

If you have already signed in, please refresh the page.

from answercheck import checkanswer

checkanswer(y_np,'993d5cbc6ddeb10776ed48159780a5d3');

from answercheck import checkanswer

checkanswer(y_np,'993d5cbc6ddeb10776ed48159780a5d3');

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.

If you have already signed in, please refresh the page.

x = [-1, 0, 2, 3.1]
y = x.copy()
y[2] = 20.2

x_np = np.array(x)
y_np = np.array(y)

x = [-1, 0, 2, 3.1]
y = x.copy()
y[2] = 20.2

x_np = np.array(x)
y_np = np.array(y)

If you have already signed in, please refresh the page.

x == y

x == y

False

If you have already signed in, please refresh the page.

np.array(x_np) == np.array(y_np)

np.array(x_np) == np.array(y_np)

array([ True,  True, False,  True])

Zero vectors and Ones vectors in numpy

zeros(n) creates a vector with all 0s
ones(n) creates a vector with all 1s

Do This

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.

If you have already signed in, please refresh the page.

x_np = np.array([-1, 0, 2, 3.1])
### Define zero_np and ones_np here

x_np = np.array([-1, 0, 2, 3.1])
### Define zero_np and ones_np here

If you have already signed in, please refresh the page.

from answercheck import checkanswer

checkanswer([zero_np, ones_np],'7f874c2e446786655ff96f0bbae8a0c6');

from answercheck import checkanswer

checkanswer([zero_np, ones_np],'7f874c2e446786655ff96f0bbae8a0c6');

Random vectors

random.random(n) creates a random vector with dimension $n$ .

If you have already signed in, please refresh the page.

random_np = np.random.random(2)
print(random_np)

random_np = np.random.random(2)
print(random_np)

[0.44542519 0.09046139]

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.

If you have already signed in, please refresh the page.

x_np = np.array([1,2,3])
y_np = np.array([100,200,300])

v_sum = x_np + y_np
v_diff = x_np - y_np

print (f'Sum of vectors: {v_sum}')
print (f'Difference of vectors: {v_diff}')

x_np = np.array([1,2,3])
y_np = np.array([100,200,300])

v_sum = x_np + y_np
v_diff = x_np - y_np

print (f'Sum of vectors: {v_sum}')
print (f'Difference of vectors: {v_diff}')

Sum of vectors: [101 202 303]
Difference of vectors: [ -99 -198 -297]

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.

Do This

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:

If you have already signed in, please refresh the page.

x = [1,2,3]
y = [100,200,300]

v_sum = x + y
v_diff = x - y

print (f'Sum of vectors: {v_sum}')
print (f'Difference of vectors: {v_diff}')

x = [1,2,3]
y = [100,200,300]

v_sum = x + y
v_diff = x - y

print (f'Sum of vectors: {v_sum}')
print (f'Difference of vectors: {v_diff}')

Scalar-vector addition

A scalar-vector addition means that the scalar (or a 1-vector) is added to all elements of the vector.

Do This

Add a scalar 20.20 to all elements of the following vector x_np and store teh result back into x_np

If you have already signed in, please refresh the page.

x_np = np.array([1.0,2.0,3.0])

x_np = np.array([1.0,2.0,3.0])

If you have already signed in, please refresh the page.

from answercheck import checkanswer

checkanswer(x_np,'2f8cbcce405fa12b8608422ff28544bb');

from answercheck import checkanswer

checkanswer(x_np,'2f8cbcce405fa12b8608422ff28544bb');

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)

Do This

Divide all elements of the following vector x_np by 20.20 and put it into y_np

If you have already signed in, please refresh the page.

x_np = np.array([1,2,3])
#####Start your code here #####
y_np = 
#####End of your code here#####
print(y_np)

x_np = np.array([1,2,3])
#####Start your code here #####
y_np = 
#####End of your code here#####
print(y_np)

If you have already signed in, please refresh the page.

from answercheck import checkanswer

checkanswer(y_np,'90c1b8639f9d350af1d971d89209a0c6');

from answercheck import checkanswer

checkanswer(y_np,'90c1b8639f9d350af1d971d89209a0c6');

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 -

Note

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

Do This

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).

If you have already signed in, please refresh the page.

p_initial = np.array([22.15, 89.32, 56.77])
p_final = np.array([23.05, 87.32, 53.13])

p_initial = np.array([22.15, 89.32, 56.77])
p_final = np.array([23.05, 87.32, 53.13])

If you have already signed in, please refresh the page.

from answercheck import checkanswer

checkanswer(r,'0e231e6cfbef65cf178208cf377af85c');

from answercheck import checkanswer

checkanswer(r,'0e231e6cfbef65cf178208cf377af85c');

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.

If you have already signed in, please refresh the page.

x_np = np.array([1,2])
y_np = np.array([3,4])
alpha = 0.5
beta = -0.8
c = alpha*x_np + beta*y_np
print(c)

x_np = np.array([1,2])
y_np = np.array([3,4])
alpha = 0.5
beta = -0.8
c = alpha*x_np + beta*y_np
print(c)

We can also define a function lincomb to performn the linear combination.

Do This

Finish the following code for lincomb and compare the results we just get.

If you have already signed in, please refresh the page.

def lincomb(coef, vectors): 
    n = len(vectors[0])  # get the dimension of the vectors. note they have to be of the same dimension
    comb = np.zeros(n)   # initial the value with all zeros.
    ### Add code here to calculate the linear combination of the input vecotrs and the coefficients. 
    return comb

def lincomb(coef, vectors): 
    n = len(vectors[0])  # get the dimension of the vectors. note they have to be of the same dimension
    comb = np.zeros(n)   # initial the value with all zeros.
    ### Add code here to calculate the linear combination of the input vecotrs and the coefficients. 
    return comb

If you have already signed in, please refresh the page.

from answercheck import checkanswer

combination = lincomb([alpha, beta], [x_np,y_np])

checkanswer(combination,'8bab7329c94f3e3cda423add411685c2');

from answercheck import checkanswer

combination = lincomb([alpha, beta], [x_np,y_np])

checkanswer(combination,'8bab7329c94f3e3cda423add411685c2');

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 have already signed in, please refresh the page.

combination = lincomb([0, 0], [x_np,y_np])

combination == np.zeros(combination.shape)

combination = lincomb([0, 0], [x_np,y_np])

combination == np.zeros(combination.shape)

If you have already signed in, please refresh the page.

combination = lincomb([2, 2], [combination,np.zeros(combination.shape)])

combination == 2*combination

combination = lincomb([2, 2], [combination,np.zeros(combination.shape)])

combination == 2*combination

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.

If you have already signed in, please refresh the page.

combination = lincomb([2, 2], [combination,np.zeros(combination.shape)])

np.alltrue(combination == 2*combination)

combination = lincomb([2, 2], [combination,np.zeros(combination.shape)])

np.alltrue(combination == 2*combination)

SyntaxError

Do This

Do This

Do This

Note

Numpy

Scalars versus 1-vectors

Lists of vectors

Do This

Indexing

Do This

Do This

Do This

Assignment versus copying

Do This

Vector equality in numpy and list

Zero vectors and Ones vectors in numpy

Do This

Random vectors

Vector addition and subtraction

Do This

Scalar-vector addition

Do This

Scalar-vector multiplication and division

Do This

Element-wise operations

Note

Do This

Linear combination

Do This

Support Center

How can we help?