Skip to content
Related Articles

Related Articles

Python | Filter out integers from float numpy array
  • Last Updated : 15 Mar, 2019
GeeksforGeeks - Summer Carnival Banner

Given a numpy array, the task is to filter out integers from an array containing float and integers. Let’s see few methods to solve a given task.

Method #1 : Using astype(int)




# Python code to demonstrate 
# filtering integers from numpy array
# containing integers and float
  
import numpy as np
  
# initialising array
ini_array = np.array([1.0, 1.2, 2.2, 2.0, 3.0, 2.0])
  
# printing initial array
print ("initial array : ", str(ini_array))
  
# filtering integers
result = ini_array[ini_array != ini_array.astype(int)]
  
# printig resultant
print ("final array", result)
Output:
initial array :  [ 1.   1.2  2.2  2.   3.   2. ]
final array [ 1.2  2.2]

 
Method #2: Using np.equal() and np.mod()




# Python code to demonstrate 
# filtering integers from numpy array
# containing integers and float
  
import numpy as np
  
# initialising array
ini_array = np.array([1.0, 1.2, 2.2, 2.0, 3.0, 2.0])
  
# printing initial array
print ("initial array : ", str(ini_array))
  
# filtering integers
result = ini_array[~np.equal(np.mod(ini_array, 1), 0)]
  
# printing resultant
print ("final array : ", str(result))
Output:



initial array :  [ 1.   1.2  2.2  2.   3.   2. ]
final array :  [ 1.2  2.2]

 
Method #3: Using np.isclose()




# Python code to demonstrate 
# filtering integers from numpy array
# containing integers and float
  
import numpy as np
  
# initialising array
ini_array = np.array([1.0, 1.2, 2.2, 2.0, 3.0, 2.0])
  
# printing initial array
print ("initial array : ", str(ini_array))
  
# filtering integers
mask = np.isclose(ini_array, ini_array.astype(int))
result = ini_array[~mask]
  
# printing resultant
print ("final array : ", str(result))
Output:
initial array :  [ 1.   1.2  2.2  2.   3.   2. ]
final array :  [ 1.2  2.2]

Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course.

My Personal Notes arrow_drop_up
Recommended Articles
Page :