Array Slicing using NumPy

When I started working on a new classification project, I had to slice data array to get the desired result. That’s when I realized I have not understood array slicing completely. That prompted me to get my hands dirty on array slicing and I’ll share my learning in this post.

We need to use numpy library to play around arrays. So I’ll first import numpy.

import numpy as np

Next, I’ll create a 2-d array with random numbers.

arr2d = np.random.randint(10, size=(4, 5))

This is how the array looks like

array([[3, 7, 3, 2, 0],
       [8, 1, 6, 1, 9],
       [3, 3, 3, 8, 2],
       [5, 9, 5, 1, 3]])

Essentially, there are two parts while describing this array.

arr2d[rowFrom:rowTo-1, columnFrom:columnTo-1]

The first section tells numpy how many (or which) rows we are interested in, and the second section tells numpy how many (or which) columns we are interested in.

If I don’t mention any values in those sections like this:

arr2d[:,:]

It shows me the complete array.

array([[3, 7, 3, 2, 0],
       [8, 1, 6, 1, 9],
       [3, 3, 3, 8, 2],
       [5, 9, 5, 1, 3]])

Let’s experiment on the rows first. In this example

arr2d[:2,:]

It returns first 2 rows. Do note that if rowFrom is blank, it means it will fetch from the start (0th element).

array([[3, 7, 3, 2, 0],
       [8, 1, 6, 1, 9]])

Now I’ll specify rowFrom as well as rowTo.

arr2d[2:4,:]

Do note that, since I specified rowTo as 4, essentially it returns from index 2 to index 4-1 which is 3.

array([[3, 3, 3, 8, 2],
       [5, 9, 5, 1, 3]])

Another example.

arr2d[3:4,:]

This will get me the rows from index 3 to index (4-1) which is 3 itself. So i’ll get only one row.

array([[5, 9, 5, 1, 3]])

Similarly, we can specify the columns for array.

So, arr2d[:,:] is same as arr2d[:,0:5] which will get me all the columns.

array([[3, 7, 3, 2, 0],
       [8, 1, 6, 1, 9],
       [3, 3, 3, 8, 2],
       [5, 9, 5, 1, 3]])

Another example, this command will get me columns from 0th column to (4-1) that is 3rd column.

arr2d[:,0:4]
array([[3, 7, 3, 2],
       [8, 1, 6, 1],
       [3, 3, 3, 8],
       [5, 9, 5, 1]])

This command will get me columns from 2nd (index) column to 3rd (4-1) column.

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

What if I don’t want a specific column? There are two ways to achieve it (as far as my knowledge goes at this point of time).

Method 1: Specify the columns index which are required.
arr2d[:,[0,2,3,4]]

The output does not contain index 1 column.

array([[3, 3, 2, 0],
       [8, 6, 1, 9],
       [3, 3, 8, 2],
       [5, 5, 1, 3]])

Even though the above method works as expected, sometimes it may not be practical to specify all the columns just to remove one column, especially when we are dealing with large array.

Luckily, NumPy provides another method like shown below.

numpy.delete(arr2d,[1],1)

Through Delete method, we can delete a specific column from the array. So the result will be like this. The second parameter is axis. Since it is a 2-d array, we need to specify 1.

array([[3, 3, 2, 0],
       [8, 6, 1, 9],
       [3, 3, 8, 2],
       [5, 5, 1, 3]])

This set of exercises helped me understand how array slicing can be done. With this understanding, I’ll go ahead and work on my next project.

Comments

Post a Comment

Popular posts from this blog

Ordinal Encoder, OneHotEncoder and LabelBinarizer in Python

Usually in the data set we work on, we will have both numerical data as well as categorical data (non-numeric). There are algorithms which do not work on categorical data. These algorithms are designed that way to enhance efficiency. So in such cases how do we train the model with categorical data? This is where we need to apply a technique called Encoding. Encoding transforms categorical data into numeric. There are various Encoders available in sci-kit learn library. In this post, I'll share my learning on three Encoders. Ordinal Encoder This performs ordinal (integer) encoding of categorical data. Let's look at an example. I have an array of categorical data. This is a 2 dimensional array. multiArray = np.array([[ 'Karnataka' ,  'KA' ], [ 'Maharastra' ,  'MH' ], [ 'Gujarat' ,  'GJ' ]]) Then I use OrdinalEncoder to transform this data. ordinalEncoder = OrdinalEncoder() ordinalEncoderArray = ordinalEncoder.fit_transform(mult...
Read more

Data Visualization using Pandas - Univariate Plots

Image
Data visualization is an important step in ML. That helps us to understand the data in visual representation. Why does it matter? Because it helps us to understand the distribution of the data, any outliers, and also relationship between multiple attributes. Experts also mention that understanding the distribution helps us in choosing correct algorithm. I cannot comment on this statement as of now, as I do not have enough knowledge on that. When I reach that stage, I’ll definitely share. Pandas library provides methods to plot data in multiple ways. In fact, I did use one of those plots called  Box and Whisker Plots  in my previous examples. However, those plots were used to see the difference between multiple algorithms. In this post, I’ll share how to visualize the data which is an important step much before we apply any ML algorithm. I’ll use the data set which was used for  Binary Classification . The data set is   Pima Indians Diabetes Database  which ...
Read more

Stemming in Python

While working on NLP, we come across huge volume of words. Before using these words to the ML algorithm, a lot of pre-processing happens. These pre-processing ensures we keep relevant words which can then be used to train an ML algorithm. One such technique is called Stemming . As defined in Wikipedia - "In linguistics, a stem is a part of a word used with slightly different meanings and would depend on the morphology of the language in question." A simple example is the words approximation , approximate , approximated and approximately have the same root. So as part of Stemming, we need to make sure these words should be taken as a single word instead of four different words. So how do we achieve this in Python? The NLTK library provides support for Stemming. There are two algorithms for stemming - PorterStemmer and LancasterStemmer . I'll not get into the details of how these two works, but at a high level LancasterStemmer does a lot more iterations compared to Porte...
Read more