NumPy: Remove the leading and trailing whitespaces of all the elements of a given array

Last update on December 21 2024 07:39:56 (UTC/GMT +8 hours)

Write a NumPy program to remove the leading and trailing whitespaces of all the elements of a given array.

Sample Solution:

Python Code:

# Importing necessary library
import numpy as np

# Creating a NumPy array containing strings with leading and trailing whitespaces
x = np.array([' python exercises ', ' PHP  ', ' java  ', '  C++'], dtype=np.str)

# Displaying the original array
print("Original Array:")
print(x)

# Removing the leading and trailing whitespaces from the strings
stripped = np.char.strip(x)

# Displaying the array after removing leading and trailing whitespaces
print("\nRemove the leading and trailing whitespaces: ", stripped) 

Sample Input:

([' python exercises ', ' PHP  ', ' java  ', '  C++'], dtype=np.str)

Sample Output:

Original Array:
[' python exercises ' ' PHP  ' ' java  ' '  C++']

Remove the leading and trailing whitespaces:  ['python exercises' 'PHP' 'java' 'C++']

Explanation:

In the above exercise –

x = np.array([' python exercises ', ' PHP ', ' java ', ' C++'], dtype=np.str): This line creates a one dimensional NumPy array x with 4 elements of string data type.

stripped = np.char.strip(x): This function removes leading and trailing whitespace characters from each element of the array and returns a new array with the stripped strings. The resulting stripped array is then stored in the ‘stripped’ variable.

Pictorial Presentation:

Python-Numpy Code Editor: