Last Updated on June 13, 2020

Machine learning data is represented as arrays.

In Python, data is almost universally represented as NumPy arrays.

If you are new to Python, you may be confused by some of the pythonic ways of accessing data, such as negative indexing and array slicing.

In this tutorial, you will discover how to manipulate and access your data correctly in NumPy arrays.

After completing this tutorial, you will know:

- How to convert your list data to NumPy arrays.
- How to access data using Pythonic indexing and slicing.
- How to resize your data to meet the expectations of some machine learning APIs.

**Kick-start your project** with my new book Linear Algebra for Machine Learning, including *step-by-step tutorials* and the *Python source code* files for all examples.

Let’s get started.

**Update Jul/2019**: Fixed small typo related to reshaping 1D data (thanks Rodrigue).

## Tutorial Overview

This tutorial is divided into 4 parts; they are:

- From List to Arrays
- Array Indexing
- Array Slicing
- Array Reshaping

### Need help with Linear Algebra for Machine Learning?

Take my free 7-day email crash course now (with sample code).

Click to sign-up and also get a free PDF Ebook version of the course.

## 1. From List to Arrays

In general, I recommend loading your data from file using Pandas or even NumPy functions.

For examples, see the post:

This section assumes you have loaded or generated your data by other means and it is now represented using Python lists.

Let’s look at converting your data in lists to NumPy arrays.

### One-Dimensional List to Array

You may load your data or generate your data and have access to it as a list.

You can convert a one-dimensional list of data to an array by calling the array() NumPy function.

1 2 3 4 5 6 7 8 |
# one dimensional example from numpy import array # list of data data = [11, 22, 33, 44, 55] # array of data data = array(data) print(data) print(type(data)) |

Running the example converts the one-dimensional list to a NumPy array.

1 2 |
[11 22 33 44 55] <class 'numpy.ndarray'> |

### Two-Dimensional List of Lists to Array

It is more likely in machine learning that you will have two-dimensional data.

That is a table of data where each row represents a new observation and each column a new feature.

Perhaps you generated the data or loaded it using custom code and now you have a list of lists. Each list represents a new observation.

You can convert your list of lists to a NumPy array the same way as above, by calling the array() function.

1 2 3 4 5 6 7 8 9 10 |
# two dimensional example from numpy import array # list of data data = [[11, 22], [33, 44], [55, 66]] # array of data data = array(data) print(data) print(type(data)) |

Running the example shows the data successfully converted.

1 2 3 4 |
[[11 22] [33 44] [55 66]] <class 'numpy.ndarray'> |

## 2. Array Indexing

Once your data is represented using a NumPy array, you can access it using indexing.

Let’s look at some examples of accessing data via indexing.

### One-Dimensional Indexing

Generally, indexing works just like you would expect from your experience with other programming languages, like Java, C#, and C++.

For example, you can access elements using the bracket operator [] specifying the zero-offset index for the value to retrieve.

1 2 3 4 5 6 7 |
# simple indexing from numpy import array # define array data = array([11, 22, 33, 44, 55]) # index data print(data[0]) print(data[4]) |

Running the example prints the first and last values in the array.

1 2 |
11 55 |

Specifying integers too large for the bound of the array will cause an error.

1 2 3 4 5 6 |
# simple indexing from numpy import array # define array data = array([11, 22, 33, 44, 55]) # index data print(data[5]) |

Running the example prints the following error:

1 |
IndexError: index 5 is out of bounds for axis 0 with size 5 |

One key difference is that you can use negative indexes to retrieve values offset from the end of the array.

For example, the index -1 refers to the last item in the array. The index -2 returns the second last item all the way back to -5 for the first item in the current example.

1 2 3 4 5 6 7 |
# simple indexing from numpy import array # define array data = array([11, 22, 33, 44, 55]) # index data print(data[-1]) print(data[-5]) |

Running the example prints the last and first items in the array.

1 2 |
55 11 |

### Two-Dimensional Indexing

Indexing two-dimensional data is similar to indexing one-dimensional data, except that a comma is used to separate the index for each dimension.

1 |
data[0,0] |

This is different from C-based languages where a separate bracket operator is used for each dimension.

1 |
data[0][0] |

For example, we can access the first row and the first column as follows:

1 2 3 4 5 6 |
# 2d indexing from numpy import array # define array data = array([[11, 22], [33, 44], [55, 66]]) # index data print(data[0,0]) |

Running the example prints the first item in the dataset.

1 |
11 |

If we are interested in all items in the first row, we could leave the second dimension index empty, for example:

1 2 3 4 5 6 |
# 2d indexing from numpy import array # define array data = array([[11, 22], [33, 44], [55, 66]]) # index data print(data[0,]) |

This prints the first row of data.

1 |
[11 22] |

## 3. Array Slicing

So far, so good; creating and indexing arrays looks familiar.

Now we come to array slicing, and this is one feature that causes problems for beginners to Python and NumPy arrays.

Structures like lists and NumPy arrays can be sliced. This means that a subsequence of the structure can be indexed and retrieved.

This is most useful in machine learning when specifying input variables and output variables, or splitting training rows from testing rows.

Slicing is specified using the colon operator ‘:’ with a ‘*from*‘ and ‘*to*‘ index before and after the column respectively. The slice extends from the ‘from’ index and ends one item before the ‘to’ index.

1 |
data[from:to] |

Let’s work through some examples.

### One-Dimensional Slicing

You can access all data in an array dimension by specifying the slice ‘:’ with no indexes.

1 2 3 4 5 |
# simple slicing from numpy import array # define array data = array([11, 22, 33, 44, 55]) print(data[:]) |

Running the example prints all elements in the array.

1 |
[11 22 33 44 55] |

The first item of the array can be sliced by specifying a slice that starts at index 0 and ends at index 1 (one item before the ‘to’ index).

1 2 3 4 5 |
# simple slicing from numpy import array # define array data = array([11, 22, 33, 44, 55]) print(data[0:1]) |

Running the example returns a subarray with the first element.

1 |
[11] |

We can also use negative indexes in slices. For example, we can slice the last two items in the list by starting the slice at -2 (the second last item) and not specifying a ‘to’ index; that takes the slice to the end of the dimension.

1 2 3 4 5 |
# simple slicing from numpy import array # define array data = array([11, 22, 33, 44, 55]) print(data[-2:]) |

Running the example returns a subarray with the last two items only.

1 |
[44 55] |

### Two-Dimensional Slicing

Let’s look at the two examples of two-dimensional slicing you are most likely to use in machine learning.

#### Split Input and Output Features

It is common to split your loaded data into input variables (X) and the output variable (y).

We can do this by slicing all rows and all columns up to, but before the last column, then separately indexing the last column.

For the input features, we can select all rows and all columns except the last one by specifying ‘:’ for in the rows index, and :-1 in the columns index.

1 |
X = [:, :-1] |

For the output column, we can select all rows again using ‘:’ and index just the last column by specifying the -1 index.

1 |
y = [:, -1] |

Putting all of this together, we can separate a 3-column 2D dataset into input and output data as follows:

1 2 3 4 5 6 7 8 9 10 |
# split input and output from numpy import array # define array data = array([[11, 22, 33], [44, 55, 66], [77, 88, 99]]) # separate data X, y = data[:, :-1], data[:, -1] print(X) print(y) |

Running the example prints the separated X and y elements. Note that X is a 2D array and y is a 1D array.

1 2 3 4 |
[[11 22] [44 55] [77 88]] [33 66 99] |

#### Split Train and Test Rows

It is common to split a loaded dataset into separate train and test sets.

This is a splitting of rows where some portion will be used to train the model and the remaining portion will be used to estimate the skill of the trained model.

This would involve slicing all columns by specifying ‘:’ in the second dimension index. The training dataset would be all rows from the beginning to the split point.

1 2 |
dataset train = data[:split, :] |

The test dataset would be all rows starting from the split point to the end of the dimension.

1 |
test = data[split:, :] |

Putting all of this together, we can split the dataset at the contrived split point of 2.

1 2 3 4 5 6 7 8 9 10 11 |
# split train and test from numpy import array # define array data = array([[11, 22, 33], [44, 55, 66], [77, 88, 99]]) # separate data split = 2 train,test = data[:split,:],data[split:,:] print(train) print(test) |

Running the example selects the first two rows for training and the last row for the test set.

1 2 3 |
[[11 22 33] [44 55 66]] [[77 88 99]] |

## 4. Array Reshaping

After slicing your data, you may need to reshape it.

For example, some libraries, such as scikit-learn, may require that a one-dimensional array of output variables (y) be shaped as a two-dimensional array with one column and outcomes for each row.

Some algorithms, like the Long Short-Term Memory recurrent neural network in Keras, require input to be specified as a three-dimensional array comprised of samples, timesteps, and features.

It is important to know how to reshape your NumPy arrays so that your data meets the expectation of specific Python libraries. We will look at these two examples.

### Data Shape

NumPy arrays have a shape attribute that returns a tuple of the length of each dimension of the array.

For example:

1 2 3 4 5 |
# array shape from numpy import array # define array data = array([11, 22, 33, 44, 55]) print(data.shape) |

Running the example prints a tuple for the one dimension.

1 |
(5,) |

A tuple with two lengths is returned for a two-dimensional array.

1 2 3 4 5 6 7 8 9 |
# array shape from numpy import array # list of data data = [[11, 22], [33, 44], [55, 66]] # array of data data = array(data) print(data.shape) |

Running the example returns a tuple with the number of rows and columns.

1 |
(3, 2) |

You can use the size of your array dimensions in the shape dimension, such as specifying parameters.

The elements of the tuple can be accessed just like an array, with the 0th index for the number of rows and the 1st index for the number of columns. For example:

1 2 3 4 5 6 7 8 9 10 |
# array shape from numpy import array # list of data data = [[11, 22], [33, 44], [55, 66]] # array of data data = array(data) print('Rows: %d' % data.shape[0]) print('Cols: %d' % data.shape[1]) |

Running the example accesses the specific size of each dimension.

1 2 |
Rows: 3 Cols: 2 |

### Reshape 1D to 2D Array

It is common to need to reshape a one-dimensional array into a two-dimensional array with one column and multiple rows.

NumPy provides the reshape() function on the NumPy array object that can be used to reshape the data.

The reshape() function takes a single argument that specifies the new shape of the array. In the case of reshaping a one-dimensional array into a two-dimensional array with one column, the tuple would be the shape of the array as the first dimension (data.shape[0]) and 1 for the second dimension.

1 |
data = data.reshape((data.shape[0], 1)) |

Putting this all together, we get the following worked example.

1 2 3 4 5 6 7 8 9 |
# reshape 1D array from numpy import array from numpy import reshape # define array data = array([11, 22, 33, 44, 55]) print(data.shape) # reshape data = data.reshape((data.shape[0], 1)) print(data.shape) |

Running the example prints the shape of the one-dimensional array, reshapes the array to have 5 rows with 1 column, then prints this new shape.

1 2 |
(5,) (5, 1) |

### Reshape 2D to 3D Array

It is common to need to reshape two-dimensional data where each row represents a sequence into a three-dimensional array for algorithms that expect multiple samples of one or more time steps and one or more features.

A good example is the LSTM recurrent neural network model in the Keras deep learning library.

The reshape function can be used directly, specifying the new dimensionality. This is clear with an example where each sequence has multiple time steps with one observation (feature) at each time step.

We can use the sizes in the shape attribute on the array to specify the number of samples (rows) and columns (time steps) and fix the number of features at 1.

1 |
data.reshape((data.shape[0], data.shape[1], 1)) |

Putting this all together, we get the following worked example.

1 2 3 4 5 6 7 8 9 10 11 12 |
# reshape 2D array from numpy import array # list of data data = [[11, 22], [33, 44], [55, 66]] # array of data data = array(data) print(data.shape) # reshape data = data.reshape((data.shape[0], data.shape[1], 1)) print(data.shape) |

Running the example first prints the size of each dimension in the 2D array, reshapes the array, then summarizes the shape of the new 3D array.

1 2 |
(3, 2) (3, 2, 1) |

## Further Reading

This section provides more resources on the topic if you are looking go deeper.

- An Informal Introduction to Python
- Array Creation in NumPy API
- Indexing and Slicing in NumPy API
- Basic Indexing in NumPy API
- NumPy shape attribute
- NumPy reshape() function

## Summary

In this tutorial, you discovered how to access and reshape data in NumPy arrays with Python.

Specifically, you learned:

- How to convert your list data to NumPy arrays.
- How to access data using Pythonic indexing and slicing.
- How to resize your data to meet the expectations of some machine learning APIs.

Do you have any questions?

Ask your questions in the comments below and I will do my best to answer.

Great articles. Still get much base knowledge though used python for a while.

I’m glad it helped!

I have a clarification regarding converting single dimensional array to matrix form(2D). Why we have to convert 1D to 2D in machine learning?

Most machine learning algorithms expect a matrix as input, each row is one observation.

Dear Dr Jason, :,

Thank you for providing this tutorial on slicing. It will certainly help me understand material on LSTMs on your page, and your e-books.

My question is on 2-D slicing and the method of slicing. I would like a “generalised” concept of slicing.

In 1-D slicing an array can be split as:

myarray[from:to] – that is understood

In 2-D slicing, “Split Input and Output Features”, you gave two examples of splitting

x = [ : , :-1]

y = [ : , -1]

I would like clarification please in the context of myarray[from:to] especially for 2D splicing especially where there are two colons ‘:’

myarray[from: to] and x [ : , : -1] For x, to = : , and from = : – 1,

Thank you,

Anthony of NSW

Good question Anthony!

When you just provide an index instead of a slice, like -1, it will select only that column/row, in this case the last column.

Doers that help?

Hello;

thanks for such nice tutorial, i am new to numpy. While I am working on deep learning algorithms at a certain juncture in my program I need to create an array of (1000,256,256,3) dims where 1000 images data (of size 256*256*3) can be loaded.

Can you kindly help me in doing this in Python.

code looks like the following:

x_train=np.empty([256,256,3])

for i in range(0,1000):

pic= Image.open(f'{path}%s.jpg’ %i)

pic=pic.resize([256,256])

x_train2= np.asarray(pic)

x_train=np.stack((x_train,x_train2))

print(x_train.shape)

Thanks

A good trick is to load the data into a list, convert the list to an array, and then reshape the array to the required dimensionality.

Great. the best description I had seen.

Thanks, I’m glad it helped.

Hi Jason,

How to convert a 3D array into 1D array..

Help me please..

X = X.reshape((X.shape[0], X.shappe[1], 1))

X=X.reshape((X.shape[0]*X.shape[1]*X.shape[2],1))

We can access 2D array just like C: data[0][0].

Using data[0, 0] is not the only way like you said.

Which version of python are you using?

Using [i, j] is valid for 2d numpy array access in Python 2 and 3.

can you help me with this sir..your help will be much apreciated. thanks in advance!

https://stackoverflow.com/questions/55645616/how-to-output-masks-using-vis-util-visualize-boxes-and-labels-on-image-array

Perhaps you can summarize the issue for me in a sentence or two?

I’ve been searching for this information for so long. Thank you so so so much sir

You’re welcome!

Jason, you help me a lot

I’m happy to hear that it helped.

Thank you so much for this clear tutorial.

I’m glad it helped.

Hi, I am dealing with normalization of 3d timeseries data.

I have 3d time series data (samples, features, timestemps).

How am I supposed to reshape my 3d data to do normalization?

I am quessing it goes like this:

From [samples, features, timesteps] to ([timesteps,features] or [features,timesteps]) and then use like MinMaxScaler to fit_transform on the training data and then transform on test data?

It is a good idea to normalize per time series across all samples.

I give examples here:

https://machinelearningmastery.com/machine-learning-data-transforms-for-time-series-forecasting/

Thank you very much for this very useful, very easy to follow and understand introduction to NumPy arrays. It was very helpful.

Thanks, I’m glad it helped.

Hi jason,

my Dataframe has non – image data of dimensions (1446736, 11).

How can i reshape this array to be fed into CNN model

This may help:

https://machinelearningmastery.com/faq/single-faq/what-is-the-difference-between-samples-timesteps-and-features-for-lstm-input

thanks Jason will this apply to CNN model as well as this for LSTM ryt ?

Yes.

Hi Jason, thanks for the tutorial, really helps strengthening the basics.

I have a somewhat related question – the numpy reshape function has a default reshape order C. When working in Keras, is there any difference in using the numpy reshape or the keras native reshape? Uses that one the same order? And if not, is it possible to use numpy native functions directly in Keras code?

Thanks! 🙂

They solve different things.

You can use numpy to reshape your arrays.

Keras offers a reshape layer, to reshape tensors as part of a neural net.

You’re awesome dude! So well explained. Thank you!

Thanks, I’m glad it helped.

Hi,

Just wondering how I can import a dataset of 2D arrays ?

Thanks

Do you mean load from file?

This might help:

http://machinelearningmastery.com/load-machine-learning-data-python/

very thoughtful of you to give this tutorial, otherwise, it would be much harder for me to follow your machine learning tutorial, thank you so much

Thanks, I’m glad it helped.

One of the best article!!

Thanks!

sir thanks a lot for this article.

You’re welcome.

Hi Jason thanks a lot for this nice tutorial again. I have a problem with the following sentence of the section 4 about reshaping.

For example, some libraries, such as scikit-learn, may require that a one-dimensional array of output variables (y) be shaped as a two-dimensional array with one column and outcomes for each column.

I’m not getting well Isn’t?

For example, some libraries, such as scikit-learn, may require that a one-dimensional array of output variables (y) be shaped as a two-dimensional array with one column and outcomes for each “row”.

In fact I’m confused.

Thanks a lot for the work done.

Yes, I mean row. Updated.

Yes, so [1, 2, 3] is a 1D with the shape (3,) becomes [[1, 2, 3]] or one column with 3 rows shape (1,3).

Dear sir, I want to input my tfidf vector of shape 156060×15103 into LSTM layer with 150 timeseries step and 15103 features .. my lstm input should look something like

(None, 150, 15103). How can I achieve this ? Please help.

If array reshaping does not help here, please suggest any alternative way of how to create lstm layer with (None, 150, 15103) using tfidf input.

My aim is to use tfidf output as input to LSTM layer with 150 timesteps.

Please help.

Set the input_shape argument on the first LSTM layer to the desired shape.

how to ask the user to input a 3d and to solve that input as an inverse

Sorry, I don’t follow. Can you elaborate?

Hi Jason,

I am experiencing some weird indexing problem where I seemingly have the correct coordinates to call both

`cv2.rectangle()`

and`plt.Rectangle()`

but then using the same coordintes for slicing does not work, ie. y1 and y2 need to be reversed. I have an issue up on OpenCV Github but can you take a look if you have a moment? https://github.com/opencv/opencv/issues/15406Sorry to hear that, this sounds like an opencv issue, not an python array issue.

Good Luck.

I have got a .dat file that contains the matrix images is of size 784-by-1990, i.e., there are totally 1990 images, and each column of the

matrix corresponds to one image of size 28-by-28 pixels.how to visualize it.Pls help.

Perhaps this will help:

https://machinelearningmastery.com/how-to-load-and-manipulate-images-for-deep-learning-in-python-with-pil-pillow/

Great Article!

I have a data to be fed into stacked LSTM. The data of the shape (81,25,12). When I do predictions using model.predict, the output will of course be 3D again (81,25,12). But I want to plot each feature by itself. I want to convert the output back into 2D and slice each column/feature for error calculations and plotting.

How can I convert 3D back into to 2D array?

thank you so much Jason!

The output does not have to be 3d, the output can be any shape you design into your model.

Nevertheless, you can reshape an array using the reshape() function:

data = data.reshape((??))

Hi Jason,

Can you explain an array slicing like [:,:,1] for a 2D array please?

Yes, it selects all rows and column 0.

the best training in ML i have ever come across…THANK YOU

Thanks.

Dear Dr Jason,

Under the heading Two-Dimensional Slicing”

Where it says:

“For the input features, we can select all rows and columns except the last one by…”

To clarify:

“For the input features, we can select all rows and all columns except the last column by….”

Where it says:

“For the output column, we can select all rows again using “:

and index just the last column by….”

To clarify:

“For the output column, we can select all rows again using “:

and index just the last row by….”

Reason, in the last example, all columns are displayed of the last row.

Alternatively, only the last row is displayed.

Thank you,

Anthony of Sydney

Dear Dr Jason,

My further elaboration on section 2

#Select all rows and first two columns

doo[:,0:2]

array([[1, 2],

[4, 5],

[7, 8]])

The list is not exhaustible, but you get a better feel by experimenting.

Thank you,

Anthony of Sydney

Dear Dr Jason,

While the above methods look at selecting either ‘successive’ rows or columns, the finishing touch is to select particular columns or particular rows.

Still exploring the fundamentals of matrix selection,

One question please:

How if you have a 3D matrix, how to slice a matrix.

Thank you

Anthony of Sydney

It is important to get good at slicing in Python.

Well done!

Thanks.

Dear Dr Jason,

I had a go at slicing a 3D array. I was able to select particular columns and particular rows. I don’t know if 3D arrays are sliced or if there is an application for slicing 3D arrays. So here it is.

I note that the slicing techniques are not exhaustible.

Whether you call this selection or slicing depends on whether you use indices or the slicing operator “:”.

Whatever procedure, the end result is that you want to get a subset of the original data structure.

What is the application of 3D slicing and/or selection?

Thank you

Anthony of Sydney

Nice work!

Dear Dr Adrian,

I came across an array splitting with four parameters:

I understand that it substitutes a 0 at particular location

To simplify: by subsituting for a,b,c,d:

What is a, b, c, d? What is the effect of a,b,c,d for rows and columns?

Thank you,

Anthony of Sydneyhank you,

Anthony of Sydney

Yes, that is “to” and “from” for rows then columns.

Dear Dr Jason,

From experimentation, a and b means to select ath row to bth-1 row and at the same time select the remaining from cth column to cth-1 column.

To illustrate;

Suppose we did an operation slicing on doo

So if we assigned to those elements

Going back:

In sum doing slicing operations

I hope to exhaust all the possible methods of index slicing

Thank you,

Anthony of Sydney

In more advanced use case, you may find yourself needing to switch the dimensions of a certain matrix. This is often the case in machine learning applications where a certain model expects a certain shape for the inputs that is different from your dataset. NumPy’s

Yes, I think I give examples using moveaxis() with images here:

https://machinelearningmastery.com/a-gentle-introduction-to-channels-first-and-channels-last-image-formats-for-deep-learning/

Please how can I convert a 1D array to 7D array?

You can use the reshape() function to specify the dimensions of existing numpy array.

Nice workflow for Numpy slicing and reshaping………got much knowledge from ur article…and it helped me a lot in my workspace..Thanks

I’m happy to hear that!

Explained better than on Stack Overflow. Thank you, man)

Thanks!

Such a great tutorial, thanks very much for your great work. keep the good work up

Thanks, I’m happy it helps!

This tutorial really helps a lot. Thank you so much. How can I give my Input to RNN as my dataset is 3d array. for example, it is (no. of samples, no. of rows, no.of columns)? because the RNN accepts input as (samples, time steps and features) rite?

You’re welcome.

Good question, see this:

https://machinelearningmastery.com/faq/single-faq/what-is-the-difference-between-samples-timesteps-and-features-for-lstm-input

‘How to Index, Slice and Reshape NumPy Arrays for Machine Learning’

Ok, I have been at this for weeks…., going through your post and then trying to figure out how you did it, especially when the code does readily run on my system.

But this post most eloquently explained the root of the problem I have been facing over the last few weeks.

And that has to do with my lack of understanding/knowledge that different python ‘libraries’ and network models require different data formatting for input/output.

Today I have found this post which explains the fundamentals of data preparedness for beginners like myself.

Thank you very much.

I have three questions for you:

1.

Is this all in your book?

2.

Is there some way you can redesign this website so that a ‘table of content’ for this site could show on the left sidebar similar to (https://pandas.pydata.org/docs/getting_started/intro_tutorials/index.html ) starting at the beginner level? This will surely help with ease of navigation and troubleshooting. It takes me HOURS to find some of the answers that I am looking for and it’s all right under my nose.

3.

Do you have a hard copy version of your book?

Winner

I do cover the basics of array indexing/manipulation in this book:

https://machinelearningmastery.com/linear_algebra_for_machine_learning/

Great suggestion, this may help:

https://machinelearningmastery.com/start-here/

On hard copies:

https://machinelearningmastery.com/faq/single-faq/can-i-get-a-hard-copy-of-your-book

Thanks a lot Jason! This is really clear and helpful for me as a beginner. Just wanted to check if there’s a typo in this sentence: “It is common to need to reshape a one-dimensional array into a two-dimensional array with one column and multiple arrays (sic).” Do you mean *rows?

Thanks, fixed!

How to enter matrix values from the text box tkinter

What is tkinter?

What is meant is how to create numpy array and array values user intervention via graphical interface

Sorry, I don’t have any tutorials on using a graphical interface for creating numpy arrays.

A very nice tutorial. One point to improve: I’d add a few words on the train-test-split() method, or a link to some reference, like the one below:

https://realpython.com/train-test-split-python-data/

You can learn more here:

https://machinelearningmastery.com/train-test-split-for-evaluating-machine-learning-algorithms/