Introduction to NumPy Library in Python

Numerical data is the most commonly used data type in Machine Learning and Data Science. On other side: Handling large amounts of numerical calculations (which can reach billions) requires more than just standard Python methods. This is where NumPy, a popular Python library, steps in to provide efficient storage and mathematical operations for data professionals. NumPy is a core package that is used in almost all machine learning projects, making it an essential skill for data professionals to master. This article will cover frequently used Numpy functions and guide you through the installation process.

Key takeaways from this blog

  • What is NumPy, and how is it different from Python lists?
  • How can we create a Numpy array?
  • Important operations like finding Shape, Reshaping, Squeezing, Expanding, slicing, Indexing, Concatenating, Stacking, Broadcasting, and some essential Mathematical operations.

Let us start by learning more about NumPy.

What is NumPy?

In 2005, Travis Oliphant released an open-source library to perform mathematical operations on large multidimensional arrays efficiently and named it Numpy. Because of its effectiveness, it became the core library and the building block of other important Python libraries, such as Pandas, MatPlotlib, Seaborn, and Scikit-learn.

What is a Numpy Array?

Numpy allows us to store numerical values in single or multiple dimensions using arrays known as Numpy arrays. Unlike Python arrays and lists, Numpy arrays are a unique structure containing partial properties of both of them. Generally, a one-dimensional array is referred to as a vector, a two-dimensional array is a matrix, and an array with three dimensions is called a tensor (a set of matrices). Numpy arrays are given the unique name of N-dimensional arrays or ndarray.

How to read a Vector, Matrix and 3D Matrix using Numpy library in Python?

One general question is, how are these ndarrays different from Python lists or inbuilt python arrays? Let's see their difference:

Difference between Lists, Python arrays, and Numpy arrays

Lists provide a convenient way to manage different data types in a single structure, but this versatility comes at the cost of memory and computing efficiency, especially when elements are of the same data type. 

NumPy arrays mitigate this problem by storing elements of similar data types, leading to significant memory savings, mainly when dealing with large amounts of data. In contrast to lists, where each cell must store information about the type of object it holds, NumPy arrays store general information about the data type at the start, reducing the memory overhead for each element.

What is the difference between Numpy arrays and Python lists?

Python has its inbuilt arrays, but they are rigid about data types. For instance, an error will occur if you attempt to store float values in an array defined for int values. Numpy arrays, on the other hand, are much more flexible. They can automatically convert data types to ensure homogeneity. So, if you pass float values to an int-typed Numpy array, the floats will be automatically converted to ints and stored.

In numpy arrays, the data type is stored in the header. When we index a particular element of an array, the value goes from the array, and the data type goes from the header, giving us the complete value.

Advantages of Numpy over lists

  • NumPy makes it possible for element-wise operations, which is impossible in the list.
  • NumPy breaks tasks into multiple fragments and executes them parallelly.
  • NumPy is also time efficient for mathematical operations, as the graph below shows that its computing speed is the fastest. This is due to parallel computing and the base code of NumPy being partly in C, C++, and Fortran, which gives it a faster execution time.

We have seen the difference. Let us see some practical use cases of NumPy.

Use Case of Numpy in Machine Learning

Numpy is the most used Python library while building machine learning or Data Science applications. We perform mathematical analysis and calculations like finding data samples' mean, median, and variance, applying filters on features, matrix multiplication, or finding gradients. All these calculations can be done within Python, but Numpy makes the speed 50x faster. It makes Numpy the first choice for development. Some direct examples can be:

  • Numpy provides the functionality of generating random numbers, which can be used to initialize the parameters for Machine learning and Deep-learning models. These random numbers can follow particular probability distribution per our needs. For example, it can give us numbers following Normal distribution.
  • Numpy is used to apply smoothening operations on data features, making the training of machine learning models stable and efficient.
  • Numpy is used to calculate the gradient of vectors to update the parameters while iterating over the training dataset. This is used in the core part of learning while training ML models.
  • Numpy is used to perform huge matrix multiplication. The number of parameters can range in billions, and performing matrix multiplication on them can be a massive bottleneck while training DL models, but Numpy handles it very efficiently.

It holds much more potential than what we mentioned till now, making it an integral and essential library to learn about. So let's begin with the installation and know some essential supports it provides.

Installation and Import of NumPy

One can find the detailed instruction to install NumPy on all operating systems in our make your system machine learning-enabled blog. To install NumPy via Python PyPI (pip), we can use the commands below,

Python2 on terminal → pip install numpy
Python3 on terminal → pip3 install numpy
Jupyter notebook python2 → !pip install numpy

Once installed, we can import this library and use it in our codes. For example:

import numpy as np

The Numpy library is imported with a new name of "np". So in future sections, whenever we call 'np', it will indirectly refer to Numpy. Let's first learn about creating a numpy array using the numpy library, and then we will see its mathematical operations.

Creating a NumPy Array

Converting lists into a numpy array

We can convert a list, a native data structure in Python, into a numpy array using the np.array() function. For example:


array([1, 2, 3])

We can also specify the datatype inside the "np.array" function. Suppose we select a data type as "int", but the input list has float values; then, while creating an array, it will take floors of those float values, as shown in the example below. Please note the difference in dtype and the corresponding output.

np.array([1,2,3.7],dtype = int)

array([1, 2, 3])

np.array([1,2,3.7],dtype = float)

array([1., 2., 3.7])

As we discussed earlier, the NumPy array can be multidimensional, and the same can be created by passing a list of lists to the np.array() function. For example, a 2X3 numpy array can be formed as follows:


array([[1, 2, 3],
       [4, 5, 6]])

Array with a fixed number

np.full() function can create an array containing a fixed number. We provide the array's shape and the number we want to fill in that array. This method will be helpful while assigning the same value to all the parameters while training the machine learning model. Let's see one example of doing that,

np.full((2,2),5) # Shape is 2X2 and we want to fill 5 in this array 

array([[5, 5],
       [5, 5]])

There is one extra function, np.zeros(), which creates an array with all elements zero. We need to pass the shape of the array as a tuple to this function, and it will provide the array. For example:


array([[0., 0.],
       [0., 0.]])

Similarly, np.ones() will give us the array of required shapes with all elements 1. For example:

# Output:
array([1., 1., 1., 1.])

Random array generation using numpy

In most Machine Learning applications, we assign random values to the parameters and then fine tune those values based on training samples. The numpy function np.random.rand() is used to create an array with random values.These random values lie in the range of [0, 1), zero included, and 1 excluded.


array([[0.76981844, 0.56005659, 0.61075499],
       [0.2434684 , 0.8560164 , 0.22834211]])

Identity Matrix Creation using Numpy

An identity matrix is a square matrix where only diagonal elements are one, and the rest are zero. These matrices are very useful while constructing the Deep-learning architecture and can be created using np.eye(). It expects the input argument to represent the number of rows for an Identity matrix to create. The Identity matrix is square, so the number of columns will equal the number of rows. For example:

# Output:
array([[1., 0., 0., 0.],
       [0., 1., 0., 0.],
       [0., 0., 1., 0.],
       [0., 0., 0., 1.]])

We can move the diagonal of an Identity matrix upward or downward by specifying the value of k in np.eye(number of rows, k=value). If the value is positive, it moves upward; for a negative value, it moves downward. Please note that the matrices we will get with a non-zero value of k are not identity matrices. An example is shown below where the diagonal moves downward:

# Output:
array([[0., 0., 0., 0.],
       [1., 0., 0., 0.],
       [0., 1., 0., 0.],
       [0., 0., 1., 0.]])

Array elements with a constant distance

Arrays in which the difference between consecutive elements remains constant are known as an evenly-spaced arrays. We can use the np.arange() method to get an evenly spaced array like this,

np.arange(0,10,3) ## np.arange(start, end, gap)

array([0, 3, 6, 9])

np.arange(4) ## The default gap is 1 and start is 0

array([0, 1, 2, 3])
#np.arange(starting_point, end_point, step_size)
array([10, 15, 20, 25])

In the above example, please note that the endpoint is not included in our array. So 30 is not there in the array as it was our endpoint. If we want the end point too, there is an alternate function np.linspace(), but here we specify the number of elements wanted in the array instead of step size as shown:

#np.linspace(starting_point, end_point, number_of_elements)
array([10., 14., 18., 22., 26., 30.])

We have learned to make a new array with the help of different methods. Let's see how to find the shape of an already existing ndarray.

Different Dimensions of ndarray

We need to know the array's number of rows, columns, and axes to get an idea about the shape and size of the array.

Let's create an array with the name np_array, which will be directly used for explaining different functions ahead, as shown:

np_array = np.array([[10,20,30],[40,50,60]])
10    20     30
40    50     60

Shape: (2,3)  Size: 6   N-Dim: 2

Number of axes of the array

We can use the ndim attribute to get the number of axes (also known as dimensions) of an array, as shown:



# We got an output as 2 as the array has two axes.

If the array contains three dimensions, then the value will be 3. 

The shape of the array

Ndarrays support the shape attribute to get the shape of an array. It returns the result in a tuple telling the types of entries in all dimensions of a ndarray. For example, the output (2,3) states 2 types of entries on axis 0 and 3 types of entries for axis = 1.

# Output:

We can also get the size of the array, which is the multiplication of each type of axes. For that, numpy provides the size attribute. For the example, we created, the output is 6 as axis1*axis2 = 2*3 = 6

# Output:

Reshaping of ndarray

We sometimes need to re-orient the existing elements in an array without changing the values of elements, and reshaping helps us with that. Reshaping becomes an important operation to multiply two matrices if dimensions are not suitable. Let's take an example as shown below.

a = np.array([10,20,30,40,50,60])

array([10, 20, 30, 40, 50, 60])
###below is the output after reshaping we get
array([[10, 20, 30],
       [40, 50, 60]])

The input provided is the shape of the matrix we want. Please note that the matrix's size (multiplication of entries of shape) should be the same as the number of elements in the original array; otherwise, an error will occur.

## Error in reshaping to 2*5
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ValueError: cannot reshape array of size 6 into shape (2,5)

# The array a was having 6 elements which can not be filled in 2*5=10 places

In the above examples, we were aware of the whole shape of the required matrix. But sometimes, we know the one axis value and need to reshape it according to that. For that, we can give input as -1 in place of the dimension of the unknown axis.

array([[10, 20],
       [30, 40],
       [50, 60]])
array([[10, 20, 30],
       [40, 50, 60]])

Transpose of a matrix using Numpy Library

Transpose is a shaping method where the number of rows and columns is swapped. For example,


array([[10, 40],
       [20, 50],
       [30, 60]])

Flattening numpy array

We use the 'flatten' attribute to convert a multidimensional array to a one-dimensional array. A common use case for flattening can be merging multiple features before compression using PCA or auto encoder. We can use flatten() or ravel() for the same.

array1 = np_array.flatten()
array2 = np_array.ravel()
print("array shape after flatten is:",array1.shape)
print("array shape after ravel is:",array2.shape)
print("array after flatten is:",array1)
print("array after ravel is:",array1)
array shape after flatten is: (6,)
array shape after ravel is: (6,)
array after flatten is: [10 20 30 40 50 60]
array after ravel is: [10 20 30 40 50 60]

How to flatten an array using Numpy library in Python?

flatten() returns a deep copy while ravel() returns a shallow copy. A deep copy creates an entirely new ndarray, and changes made to the output will not reflect in the original array. While in shallow copy, it refers to the original memory, which means that changes made to shallow copy output will also reflect in the original array.

###below is changes made in flatten output
array1[1] = 0
[[10 20 30]
 [40 50 60]]
###below is changes made in ravel output
array2[1] = 0
[[10  0 30]
 [40 50 60]]

Expanding a NumPy array

We can use the np.expand_dims() method to extend the dimension of a numpy array. The input we need to provide is the array and axis along which we wish to expand the array. If the expansion is around rows, it will look like this: 


Squeezing a NumPy array

Use the np.squeeze() method for compressing an array. Squeezing an array means reducing its dimension along an axis. The axis we choose has a corresponding value equal to 1 in the shape tuple. If, by chance, while selecting an axis, the condition of the corresponding shape value =1 is not followed, an error will occur.

a = np.array([[[1,2,3],[4,5,6]]])
# Output:
(1, 2, 3)

# Output: We get the following error as corresponding value is 3
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<__array_function__ internals>", line 180, in squeeze
  File "/home/avisouser/.local/lib/python3.8/site-packages/numpy/core/", line 1545, in squeeze
    return squeeze(axis=axis)
ValueError: cannot select an axis to squeeze out which has size not equal to one

array([[1, 2, 3],
       [4, 5, 6]])

We have seen now to create an array and determine its shape. Now we will see how to access specific elements of an array using slicing and indexing.

Slicing and Indexing of NumPy arrays

Slicing vectors or 1D arrays

Sometimes only a part of the complete array is needed. For that, we only need to pass starting index, end index, and step size as parameters in this order: [Start Index : End Index : Step Size]. For example, we have an array of sorted elements in ascending order, and we want to get all elements apart from the largest and the smallest element, then,

array([1, 2, 3, 4, 5, 6])
Output when no step size is: ## Default step size would be 1
array([2, 3, 4, 5])

Please note that here end index is not included. Step size determines the number of elements to skip from the array in choosing the next element. For example,

Output when step size is 2:
array([2, 4])  # Here note that element 3 is skipped

Slicing and indexing of matrices or 2D arrays

In 2D arrays, two axes are present. So slicing here has to occur for both axes. Please note that this method will also work for multidimensional arrays. Indexing elements in a 2D array is the same as we do indexing in the list of lists. For example,

# Output: Here we get the elemnet from first row and first column

# Output: Here we get the elemnet from first row and third column

# Output: Here we get the elemnet from second row and third column

Let's see how we can do the slicing in the case of 2D arrays.

In the example below, slicing of ndarray along a column is performed, and all rows are chosen. Programmatically it can be done as:


Output: Here we only choose the second column becuase start index is 1 and end index is 2, but 2 is excluded
Output: Here we only choose the first row
array([[10, 20, 30]])

Output: Here we only choose the first row and second column

Slicing and indexing of 3D matrices or 3D arrays

Let's create a 3D matrix using the np.array() method and then perform slicing,

a = np.array([[[10,20],[30,40],[50,60]],# first axis array

[[70,80],[90,100],[110,120]],# second axis array
[[130,140],[150,160],[170,180]]])# third axis array

# Output:
[[[ 10  20]
  [ 30  40]
  [ 50  60]]
[[ 70  80]
  [ 90 100]
  [110 120]]
[[130 140]
  [150 160]
  [170 180]]]

Please note that the 3D matrix has an additional axis compared to the 2D matrix. The third axis determines the number of 2D matrices superimposed on one another, as shown in the figure below. So while slicing the 3D matrix, we need to mention which 2D array we want to slice.

3D matrix representation in python numpy library

As discussed in the section Slicing and indexing of matrices or 2D arrays,we take slices of each axis to get our required elements.

(3, 3, 2)

## above we see that we get a 3d matrix with a depth of 2 and x, y axis as 3.

###Inexing of array
#Output: Here we get first element for depth 1 with x and y coordinate being 0

###Slicing of array
# Output: We select first two rows of second and third array
array([[[ 70,  80],
        [ 90, 100]],
[[130, 140],
        [150, 160]]])

Flipping an array using numpy

We can use the np.flip() method to flip the array horizontally or vertically, depending on the axis.

array([[10, 20, 30],
       [40, 50, 60]])

# Output:
array([[40, 50, 60],
       [10, 20, 30]])

Stacking and Concatenating NumPy arrays

There are two ways to combine two ndarrays, Stacking and Concatenating. In stacking, the number of dimensions of the output array is more than the dimension of the input array, while in concatenation, it remains the same. For example, if we stack two 1-D arrays, we get a 2-D array, while concatenation will give a 1-D array only. In stacking, the axis along which arrays are combined should have the same size; otherwise, an error will occur. 

We can use these functions for stacking and concatenation:

  • Vertical stack (np.vstack()): Here, two arrays are combined vertically, increasing the number of rows.
  • Horizontal stack (np.hstack()): Here, two arrays are combined horizontally, increasing the number of columns.
  • np.dstack(): Here, two arrays are combined along depth and increase the array's depth. 
  • np.concatenate(): Here, we combine two arrays along a particular axis.
a = np.array([1,2,3])
b = np.array([4,5,6])
a1 = np.array([[10,20],[30,40]])
b1 = np.array([[50,60],[70,80]])

# Output:
array([[1, 2, 3],
       [4, 5, 6]])

array([1, 2, 3, 4, 5, 6])

# Output:
array([[[10, 50],
        [20, 60]],
[[30, 70],
        [40, 80]]])
# Output: Here we concatenate along row
array([1, 2, 3, 4, 5, 6])

Broadcasting in NumPy arrays

Using broadcasting, we can apply simple arithmetic operations (addition, subtraction, etc.) on numpy arrays with different shapes. It beautifully leverages the functional property of Python and internally shifts some operations into a C environment rather than using Python, making execution faster.

It becomes beneficial in two cases:

  • Multiplying a scalar with a ndarray. 
  • Increase the size of smaller ndarray for a case when two ndarrays have different dimensions. Note that the dimension, which does not match, has to be 1 for a smaller matrix, and then broadcasting would work. Otherwise, it will throw an error. For example, (5,6) ndarray can be broadcasted with (6,1).
a = np.arange(10,100,20)
b = np.array([[3],[3]])

#Output: Here we get the output when we try to add 2 different dimensional ndarrays.
array([[13, 33, 53, 73, 93],
       [13, 33, 53, 73, 93]])
# Output: Here we multiply by a scalar number for the whole matrix
array([ 20,  60, 100, 140, 180])

Here the scalar number is hypothetically stretched to match the dimensions of ndarray so that it becomes feasible for multiplication.
Unless two ndarrays have the same dimensions, their calculations would not have been feasible, but now it is possible due to broadcasting.

Mathematical Operations on NumPy array

Basic Arithmetic Operations

In standard mathematics, we apply addition, subtraction, division, etc. All this can be done for a Numpy array as well.

a = np.arange(10,100,20)
print("sum output is:",a+2)
print("subtraction output is:",a-2)
print("division output is:",a/2)

array([10, 30, 50, 70, 90])
sum output is: [12 32 52 72 92]
subtraction output is: [ 8 28 48 68 88]
division output is: [ 5. 15. 25. 35. 45.]

Mean: We can find the mean of the values present in an array using the np.mean() method. For a vector, it means taking the sum of the vector and dividing it by the length of the vector.

Median: We can find the median value of an array using the np.median () method. The median is a value that separates the higher half from the lower half of data, a population, or a probability distribution.

Standard deviation: We can find the standard deviation using np.std(). Using standard deviation, we can find how much the data samples are dispersed with respect to the mean.




Minimum: We can find the minimum element in the array using the np.min() method. The index of a minimum element can be determined using the argmin() method.

Maximum: We find the max element in the array usingthe np.max() method. The index of the maximum element can be determined using the argmax()method.

Array Sum: We can usethe sum() method to find the array sum.

# Output:

# Output:

# Output:
### In above case we determine min and max element along the column

Sorting NumPy array

Often in Data Science problems, we need to sort elements. Depending on its implementation and algorithm used, the time required for sorting can vary greatly. NumPy provides inbuilt support for various algorithms like mergesort, quicksort, time sort, etc.

a = np.array([10,40,20,500])
np.sort(a, kind='mergesort')
# Output:
array([10, 20, 400, 5000])


NumPy is a game-changer for Python developers as it enables efficient mathematical operations. This article covers the fundamentals of the NumPy library, including installation and working with ndarrays. For a more in-depth understanding, refer to the official documentation. We hope you found it informative and enjoyable.


Enjoy Learning!

More From EnjoyAlgorithms

© 2022 Code Algorithms Pvt. Ltd.

All rights reserved.