Numpy first occurrence of value greater than existing value

by

This is a little faster (and looks nicer)

np.argmax(aa>5)

Since argmax will stop at the first True (“In case of multiple occurrences of the maximum values, the indices corresponding to the first occurrence are returned.”) and doesn’t save another list.

In [2]: N = 10000

In [3]: aa = np.arange(-N,N)

In [4]: timeit np.argmax(aa>N/2)
100000 loops, best of 3: 52.3 us per loop

In [5]: timeit np.where(aa>N/2)[0][0]
10000 loops, best of 3: 141 us per loop

In [6]: timeit np.nonzero(aa>N/2)[0][0]
10000 loops, best of 3: 142 us per loop

More Related Contents:

  1. How can I implement for loop in Python?
  2. Python: Justifying NumPy array
  3. Numpy where function multiple conditions
  4. testing whether a Numpy array contains a given row
  5. Limit number of threads in numpy
  6. python numpy ValueError: operands could not be broadcast together with shapes
  7. Dropping infinite values from dataframes in pandas?
  8. How can I add new dimensions to a Numpy array?
  9. What is the inverse function of zip in python? [duplicate]
  10. In-place type conversion of a NumPy array
  11. Multiprocessing.Pool makes Numpy matrix multiplication slower
  12. Extracting specific columns in numpy array
  13. Get year, month or day from numpy datetime64
  14. more efficient way to calculate distance in numpy?
  15. Get coordinates of local maxima in 2D array above certain value
  16. How to get element-wise matrix multiplication (Hadamard product) in numpy?
  17. Finding the nearest value and return the index of array in Python
  18. 2D array indexing
  19. How to get the indices list of all NaN value in numpy array?
  20. Equivalent of `polyfit` for a 2D polynomial in Python
  21. setting spacing between grouped bar plots in matplotlib
  22. Combine 3 separate numpy arrays to an RGB image in Python
  23. Numpy Vector (N,1) dimension -> (N,) dimension conversion
  24. Custom data types in numpy arrays
  25. Python’s and Numpy’s nan and set
  26. Extract csv file specific columns to list in Python
  27. Resample time series in pandas to a weekly interval
  28. How to convert an image from np.uint16 to np.uint8?
  29. numpy.sum() giving strange results on large arrays
  30. Avoid overflow when adding numpy arrays

Leave a Comment