Numpy Split Function

Introduction

In this lab, we will cover the split() function of the char module in the Numpy library. The split() function is used to split an input string into a list of strings based on a specified separator.

VM Tips

After the VM startup is done, click the top left corner to switch to the Notebook tab to access Jupyter Notebook for practice.

Sometimes, you may need to wait a few seconds for Jupyter Notebook to finish loading. The validation of operations cannot be automated because of limitations in Jupyter Notebook.

If you face issues during learning, feel free to ask Labby. Provide feedback after the session, and we will promptly resolve the problem for you.

Skills Graph

Import Numpy Library

In order to use the numpy.char.split() function, we need to import the numpy library. We can do that using the following code:

import numpy as np

Split a Simple String using split() Function

In this step, we will split a simple string using the split() function. We will define a string input_string and pass it as an argument to the split() function. The output of the function will be a list of strings obtained by splitting the input string by whitespace.

input_string = "Hello World!"
result = np.char.split(input_string)
print("Input String: {}\nOutput String: {}".format(input_string,result))

The output of the above code will be:

Input String: Hello World!
Output String: ['Hello', 'World!']

Split a String with a Separator

In this step, we will split a string using a separator. We will define a string input_string with multiple | separators. Using the split() function we will split the string based on | as a separator and output the result list.

input_string = "apple|banana|cherry"
result = np.char.split(input_string, sep='|')
print("Input String: {}\nOutput String: {}".format(input_string,result))

The output of the above code will be:

Input String: apple|banana|cherry
Output String: ['apple', 'banana', 'cherry']

Split a String with Maximum Splits

In this step, we will split a string with a maximum number of times. We will define a string input_string which is to be split only once. Using the split() function we will split the string based on | as a separator and a maximum split limit of 1.

input_string = "apple|banana|cherry"
result = np.char.split(input_string, sep='|', maxsplit=1)
print("Input String: {}\nOutput String: {}".format(input_string,result))

The output of the above code will be:

Input String: apple|banana|cherry
Output String: ['apple', 'banana|cherry']

Split an Array of Strings

In this step, we will split an array of strings. We will define an array of strings using numpy. Using the split() function we will split the array of strings based on - as a separator and a maximum split limit of 1.

input_array = np.array(['apple-juice', 'banana-milkshake', 'cherry-smoothie'])
result = np.char.split(input_array, sep='-', maxsplit=1)
print("Input Array: \n{}\nOutput Array: {}".format(input_array,result))

The output of the above code will be:

Input Array:
['apple-juice' 'banana-milkshake' 'cherry-smoothie']
Output Array:
[list(['apple', 'juice']) list(['banana', 'milkshake']) list(['cherry', 'smoothie'])]

Summary

In this lab, we learned about the split() function of the Numpy Library. We learned how to split a string into a list of strings based on a specified separator. We also covered the use of maximum split limit while splitting a string. Finally, we covered how to split an array of strings using split() function.