Data Science Numpy Python

A Simple Walk-through with NumPy for Data Science

Pinterest LinkedIn Tumblr
numpy logo thumbnail

Nearly every scientist working in Python draws on the power of NumPy.


In today’s tutorial, we will grasp this fundamental concept of what NumPy is, or also known as Numerical Python. We will dive into how to Install this nicely made and well-written library. Also, we will see some use-cases around the library. Then finally, we will dive into some technical details on what the possibilities are when working with it.

What is Numpy about?

One of the first necessary tools that most scientists use in daily research with Python is NumPy. The library is used a lot because it provides support for large multidimensional array objects (data operations, storage as the selection grows) and it gives us various tools to work with them.

The nature of numpy
Figure 1: NumPy lies at the core of a rich ecosystem of data science libraries.

If you look at the diagram provided above, you will see that this basic library offers:

  1. A foundation for most other libraries such as Matplotlib (for creating plots),  Scipy (another scientific computing library) and other libraries like Pandas and statsmodels (Statistics),  SkLearn (Machine Learning), Scikit-Image (Image Processing).
  2. It provides a robust array type representing multidimensional datasets of many different kinds and supports arithmetic operations.

Apart from what’s listed, NumPy also provides the perfect tools to use for Linear Algebra, Fourier transform and many other everyday science and engineering tasks.

Case Studies where NumPy was Used

Here is some insight into the development of particular solutions were NumPy played a significant role:

  • First Imagery of a Black Hole
Black Hole
Figure 1: The role of NumPy in Black Hole imaging

How libraries like SciPy and Matplotlib rely on NumPy enabled the Occasion Perspective Telescope to generate the first image of a black hole. Read more.

  • Pose Estimation using Deep Learning
Figure 2: Colored dots track the positions of a racehorse’s body part
(Source: Mackenzie Mathis)

DeepLabCut makes use of NumPy for improving clinical research that includes observing animal actions for a much better understanding of motor control across varieties and timescales. Read more.

Without further ado, let’s get started with the installation process and see some basic operations using NumPy.

How to Install

Suppose you have Anaconda installed on your computer already. In that case, you may skip this process since NumPy comes pre-installed once you install Anaconda, which includes Data Science packages suitable for Linux, Windows, and macOS.

If you don’t have the library already installed or you’re not using Anaconda, then you may run the following command below.

After the command above, you should see Successfully installed.

To verify the installation was complete, open up a Python Script, name it and execute the following command. You will surely get the output if everything went alright with the installation.

Creating a NumPy Array

There are numerous ways to create an empty array using NumPy. Let’s cover a few here and see the full list of options you have here.

Basic multi-dimensional array

Creating a NumPy array is very simple. All we need to do is pass the array’s values as a list into the np.array() method.

Once you check the type of the array stored in the variable, you will notice it’s a list.

After passing the list into the np.array() method, the data type automatically changes. It converts a Python List into a NumPy array. You might be wondering: what’s the difference between a NumPy array and a Python list? To know more, I have provided the answer to this question towards the end of this tutorial. For now, knowing that they’re not the same is enough.

To check the data type of the NumPy array elements stored in the variable “data”, you will notice it’s a 32 bits float. For more review on Python’s different data types, be sure to visit this tutorial were I talked about them in more detail.

Most of the time, you will want to modify the data type of the values in your NumPy array. We can easily change the data type from float32 (default) to float64.

We can even convert from floats to integers. This method in programming is usually called typecasting. The purpose of this method is to change an entity from one data type to another. However, when performing this operation, you could certainly lose information if used carelessly—for example, converting a float 2.89 to an integer 2. As you can see, information about the previous number is missing.

Array of zeros

We can also create an empty array filled with all zeros using the np.zeros() method. This method returns a new array of a given shape and fills it with zeros.

In the example, we create both a one-dimensional NumPy array 1×5 (one row, five columns) and a two-dimensional NumPy array 2×2 matrix (two rows, two columns) filled with zeros.

Array of ones

Besides creating a NumPy array filled with zeros as we have seen, we can also create a NumPy array filled with only ones this time instead of zeros using the np.ones() method.

Now let’s create a 2-dimensional NumPy array of a 3×3 matrix (three rows, three columns) filled with ones.

Random numbers in ndarrays

One commonly used method when creating n-dimensional arrays is the np.random.rand() method. This function creates an array of specific shapes (In our case, a 3×3 array) and fills it with random values from a uniform distribution between 0 and 1.

An array of your choice

Instead of creating an array of a specific size filled with random numbers within the range of zeros(0’s) – ones(1’s), we can return a variety of fill_value ( the number we want to initialize our array) with the given shape, type and order using the np.full() method.

Identity matrix in NumPy

Using NumPy, we can create a special kind of matrix called the identity matrix with its main diagonal cells all filled with ones(1’s), and the rest of the cells filled with zeros(0’s). We can create a 3×3 identity matrix using the np.eye() method. Here’s what it looks like:

And if we decide to shift the diagonal by one (1), we need to pass a value inside the parameter k. Note the zero (0) is the default parameter or also referred to as the main diagonal.

Evenly Spaced n-Dimensional Array

When working with NumPy, to generate an evenly spaced array of numbers, we can use the np.arange() method. This method returns values within a given interval. These values generated within a given interval includes the start, but excludes the stop.

Let’s see the example below. As you have noticed the value starts from zero (0) however it doesn’t end at ten (10), but instead it ends at nine(9).

Another example is specifying the start and stop value and the number of steps (the value to increment by). The default increment value is one (1).

However, NumPy also provides you with another method np.linspace() that doesn’t omit the start and stop value and generates several samples between the interval, including the start and stop value.

Checking the Shape of NumPy Array

After you have created your array using any of the options mentioned above, the next step you want to do is to check your array’s shape, as it is also essential to know when working with real world data. You might be curious to see the dimension if it’s one-dimensional, two-dimensional, and so on. Let’s see how to do this.

Dimensions of NumPy arrays

Let’s create a NumPy array and fill it manually with values. If we aren’t sure about our data dimension, we can access it simply using the .ndim function, which prints out the data dimension. In our case, it is a two-dimensional array.

Shape of NumPy array

Now let’s say we don’t know the size of the array created above, and we want to check the number of rows and columns that are in the data. We can check using the .shape function.

Size of NumPy array

Sometimes you might want to check the array’s size, meaning how many elements are inside the container. We can extract the value by merely performing a multiplication between the number of rows and columns available.

Reshaping the NumPy Array

It’s common when working with any data to change the given structure of the data. Let’s say we are given the evenly spaced n-dimensional array. We can transform a one-dimensional NumPy array into a two dimensional NumPy array using the np.reshape() function. This function takes two arguments that must be specified: the data you want to reshape and the new shape.

For example, we can reshape our data into a 3×3 matrix.

One trick I found helpful is when you are not sure about any of the shapes of the axis you want to reshape, you can just put a negative one (-1), and NumPy automatically calculates the shape for you. Let’s see some examples of this operation below.

Flattening a NumPy array

Let’s say you get a two-dimensional array as an output, and your goal is to convert the given array into a one-dimensional array. NumPy let’s us perform this operation using either the .flatten() function or the .ravel() function. Don’t get carried away by these functions.

Although they both accomplish the task, there’s a slight difference. Let’s understand these differences below.

The .ravel()

  • It returns a view of the original array, so if you modified the original array, the reference values would also change.
  • It doesn’t occupy any memory.

The .flatten()

  • It returns a copy of the original array. So if the original array is modified, the reference doesn’t get changed.
  • It occupies more memory than using the .ravel() function.

Let’s run a little experiment.

Transpose of a NumPy Array

Transposing an array is very important and also plays a major role in real life applications, for example in image processing, deep learning, machine learning methods, computer vision applications etc.

To transpose a matrix, NumPy provides a useful function that lets you perform this operation using np.tranpose() function.

As you have noticed from the output the rows and column values are swapped after the transpose operation has been performed.

If you’ve enjoyed the tutorial up until now, you should click on the “Click to Tweet Button” below to share on Twitter or simply share the link to your network. 😉

Check out a comprehensive Tutorial on NumPy Share on X

Indexing with NumPy Array

If you are familiar with how Python Lists works, then indexing will be quite simple for you to understand. If you aren’t aware, be sure to read this tutorial.

Imagine you want to get a particular element in your array. You will have to specify the i-th value (remember we start our counting from 0, not from 1) within a square bracket, which is precisely how you work with Python lists. Let’s see some examples:

Given the range from one (1) to ten (10), let’s say we want to get the value one (1). Since it’s our array’s first element, all we need to do is pass in the 0th index within the square bracket.

Passing in the index also applies to the value of five (5), which is the array’s 4th index.

Now you will notice the negative one value (-1) within the last example shown. This value gets the last element of our array if we aren’t sure about the size.

Now we know how to get to a specific value within our array. What if we want to set a new value for an individual element? We will perform this operation using the above index notation, then specify the new value. Let’s see an example.

It would help if you were careful about the data type you set when creating the array because NumPy has fixed data types. Notice the value 200.55 is now 200. Make sure you are careful about this.

The same concept applies to higher dimensions. Let’s see how we can get /set a value in a higher dimension. The only difference is we have to add a comma and the second value to specify the row and column index since it’s a two-dimensional array.

Slicing of NumPy Arrays

Now we have seen how to extract individual elements within our array, let’s see how we can get and set smaller sub-arrays within a larger array. This concept is called slicing.

To access a sub-array within a larger array, we have to use this syntax.

data[ start : stop : step-size]

As you will notice, the slice notation is marked by the colon (:) character.

It’s essential to know the step-size is set to one (1) by default, and if you want to increase the step-size, it must be specified.

Let’s see some examples of working with this.

One-dimensional subarrays

We can also choose to print out the array in reverse order by specifying a negative one (-1) within the step-size.

Two-dimensional subarrays

Once you’ve understood the concept behind working with only a one-dimensional array, let’s see how we can generalize this to working with two-dimensional images.

Three-dimensional subarrays

Now let’s see how we can work with a three-dimensional array. Note that understanding how to slice this array would surely give you a great understanding of how to crop certain parts of an RGB image containing the row, column, and depth-size.

Array Concatenation

There are multiple ways to combine or join two or more arrays into one big array in NumPy. Combining can certainly be accomplished using one of these functions:

Let’s create two arrays of the same size to see some working examples with them.


Using this function, we can vertically stack the contents of two or more arrays into a single array.


Using this function, we can horizontally stack the contents of two or more arrays into a single array.


Using this function, we can combine the contents of two or more arrays into a single array.

We can choose to perform either a row-wise concatenation by setting the axis to 0.

Or perform a column-wise concatenation by setting the axis to 1.

Broadcasting in NumPy Array

One useful function feature NumPy provides for us is the capability to perform universal binary functions (multiplication, addition, subtractions, etc.) on arrays of different sizes and shapes. This concept is known as broadcasting.

Let’s see an example using this concept. Say we want to add a scalar (1,) containing the value one (1) with a 4×4 matrix full of fives (5’s). What do you think the output will be?

Well, since it’s not the same size, it wouldn’t be possible mathematically. However, this doesn’t apply when working with NumPy arrays.

What happened here was since the scalar (1,) didn’t match the 4×4 matrix, this scalar value was padded around with ones to match the matrix before the computation.

Mathematical function with NumPy Array

NumPy provides already predefined universal functions available for you, including comparison operators, making conversions from radians to degrees, rounding and remainders, addition, subtraction, and much more. To find out more about these functions which are available to you, refer to the NumPy documentation to unveil these interesting functionalities.

Array Arithmetic with NumPy

Numpy has quite a lot of arithmetic operations. For example, the power, remainder, division, multiplication, addition, and subtraction.

Quartile, Mean, Median and Standard deviation

Once you are provided with a vast amount of data, the first action you want to perform is to compute summary statistics for the data given to you. Most of the common ones I tend to use are the mean, median, and standard deviation.

Others I find useful as well are checking the maximum and minimum value in your data.

Also, even checking the quartile provides the entire distribution of the data stored in our array.

What if you don’t want to know the minimum or the maximum value, but exactly where it is located in your array. That’s the Index. NumPy provides an aggregate function for this called np.argmin() and np.argmax()

Sorting in NumPy arrays

Sorting is an essential concept to understand in the area of engineering. Even if you are a Programmer, Data Scientist, Artificial Intelligence Engineer, etc., it’s crucial to know how these sorting algorithms work and know which one to use in a specific case with a minimum time and space complexity.

To sort an unordered array of values, NumPy lets us perform this action using the np.sort() method. You can see more details about which other options of sorting algorithms it provides in the link. Overall the default is the ‘mergesort’ algorithm.

NumPy Arrays vs Python Lists – Which is better?

If you’ve read the tutorial up to this point, you might want to know the difference between using either NumPy arrays or Python lists.

Here are the key two reasons why you should use NumPy arrays rather than Python lists.

  1. Numpy is an optimized version of Python lists. Meaning it adds support for working with multidimensional arrays and matrices along with a massive collection of high-level mathematical functions. Which can serve for:
    • Statistical analysis
    • Linear Algebra
    • Financial functions
    • Searching, Sorting, etc.
  1. Second, Numpy is written in Python and C programming languages, making it faster than Python lists which are just written in Python.


In this post, you discovered the basic concept behind NumPy and some of the most common use cases.

Do you have any questions about NumPy or this post? Leave a comment and ask your question. I’ll do my best to answer.

Further Reading

We have listed some useful resources below if you thirst for more reading.



To be notified when this next blog post goes live, be sure to enter your email address in the form!


  1. This is the most comprehensive tutorial i’ve ever read related to NumPy. Thank you David. Keep on writing.

  2. Thank you for sharing David. I love the natural flow of your content. Keep on writing.

    • David Praise Chukwuma Kalu Reply

      Thank you Mike. Much appreciated.

  3. I’ve never read a tutorial like this. Good content. Keep on writing.

  4. Thank you for the content. It’s well written and easy to understand for a begineer like me.

Write A Comment