NumPy Arrays Notes
NumPy Arrays Notes
Python Libraries
and Concepts Used
in Research
PART 2: NUMPY
Importing NumPy
import numpy as np
In [1]: np.zeros(5)
Out [1]: array([0., 0., 0., 0., 0.])
In [2]: np.zeros((5,3))
Out [2]: array([[0., 0., 0.],
[0., 0., 0.],
[0., 0., 0.],
[0., 0., 0.],
[0., 0., 0.]])
Empty array
np.empty()
does not initialize array
recommended for very large arrays where each element will be updated
2D NumPy arrays
First index row
Second index - column
2D NumPy arrays
np_array[:,j]
jth column of np_array
np_array[i,:] or np_array[i]
ith row of np_array
In [2]: z2 = z1 + 1
In [3]: z1
Out [3]: array([1, 3, 5, 7, 9])
In [4]: z2
Out [4]: array([ 2, 4, 6, 8, 10])
In [5]: ind = [0,2,3]
In [6]: z1[ind]
Out [6]: array([1, 5, 7])
In [2]: z1 > 6
Out [2]: array([False, False, False, True, True], dtype=bool)
In [3]: z1[z1 > 6]
Out [3]: array([7, 9])
In [5]: ind = z1 > 6
In [6]: ind
Out [6]: array([False, False, False, True, True], dtype=bool
In [7]: z1[ind]
Out [7]: array([7, 9])
In [2]: w
Out [2]: array([1, 3, 5])
In [3]: w[0] = 3
In [4]: w
Out [4]: array([3, 3, 5])
In [5]: z1
Out [5]: array([3, 3, 5, 7, 9])
In [2]: w = z1[ind]
In [3]: w
Out [3]: array([1, 3, 5])
In [4]: w[0] = 3
In [5]: w
Out [5]: array([3, 3, 5])
In [6]: z1
Out [6]: array([1, 3, 5, 7, 9])
In [2]: X.shape
Out [2]: (2, 3)
In [3]: X.size
Out [3]: 6
Returns a bool if all of the elements of the array fulfills the given
condition
np.all(condition)
In [1]: x = np.array(0.1,0.2,0.3,0.4,0.5,0.6,0.7,0.8,0.9,1)
In [1]:
x1 = np.array([0,1,2]); x2 = np.array([3,4,5])
In [2]:
x, y = np.meshgrid(x1, x2)
In [3]:
x
Out [3]:
array([[0, 1, 2],
[0, 1, 2],
[0, 1, 2]])
In [4]: y
Out [4]: array([[3, 3, 3],
[4, 4, 4],
[5, 5, 5]])
In [1]: x = np.arange(0,10,2)
Out [1]: array([0, 2, 4, 6, 8])