numpy remove element based on condition code example
Example: delete values with condition in numpy
a = np.array([1,2,3,4,5,6])
out = np.logical_not(a < 3)
print(out) # array([False, False, True, True, True, True])
print(a[out]) # array([3, 4, 5, 6])