Open In App

Return a boolean array which is True where the string element in array ends with suffix in Python

Improve
Improve
Like Article
Like
Save
Share
Report

In this article, we are going to see how we will return a boolean array which is True where the string element in the array ends with a suffix in Python.

numpy.char.endswith()

numpy.char.endswith() return True if the elements end with the given substring otherwise it will return False.

Syntax : np.char.endswith(input_numpy_array,’substring’)

Parameters:

  •  input_numpy_array refers to the numpy array with strings
  •  substring is compared with all elements present in an array

Return: Return the boolean array which includes “True” if a substring is present as a suffix and “False” if a substring is not present as a suffix.

Example 1: 

In this example, we are creating a NumPy array with 5 strings and checking the elements’ ends with ‘ks’.

Python3




# import numpy
import numpy as np
 
# Create 1D array of strings.
a = np.array(['hello', 'welcome to',
              'geeks', 'for', 'geeks'])
 
# check the strings in an above array
# ends with  substring - 'ks'
# and stored in a variable "gfg".
Data = np.char.endswith(a, 'ks')
 
# Print boolean array
print(Data)


Output:

[False False  True False  True]

Example 2:

In this example, we are creating a NumPy array with 5 strings and checking the element’s ends with ‘o’.

Python3




# import numpy
import numpy as np
 
# Create 1D array of strings.
a = np.array(['hello', 'welcome to',
              'geeks', 'for', 'geeks'])
 
# check the strings in an above array
# ends with  substring - 'o'
# and stored in a variable "gfg".
Data = np.char.endswith(a, 'o')
 
# Print boolean array
print(Data)


Output:

[ True  True False False False]

Time complexity: O(N), where N is the number of strings in the array “a”. 

Space complexity : O(N), as we are storing the result in a boolean array “Data” with the same number of elements as the number of strings in the array “a”.



Last Updated : 06 Feb, 2023
Like Article
Save Article
Previous
Next
Share your thoughts in the comments
Similar Reads