Replace all elements of Python NumPy Array that are greater than some value

by

I think both the fastest and most concise way to do this is to use NumPy’s built-in Fancy indexing. If you have an ndarray named arr, you can replace all elements >255 with a value x as follows:

arr[arr > 255] = x

I ran this on my machine with a 500 x 500 random matrix, replacing all values >0.5 with 5, and it took an average of 7.59ms.

In [1]: import numpy as np
In [2]: A = np.random.rand(500, 500)
In [3]: timeit A[A > 0.5] = 5
100 loops, best of 3: 7.59 ms per loop

More Related Contents:

  1. Is there a NumPy function to return the first index of something in an array?
  2. Understanding NumPy’s einsum
  3. Intuition and idea behind reshaping 4D array to 2D array in NumPy
  4. Numpy sum elements in array based on its value
  5. Efficiently return the index of the first value satisfying condition in array
  6. How to save and load numpy.array() data properly?
  7. How do I access the ith column of a NumPy multidimensional array?
  8. Interweaving two numpy arrays
  9. python how to pad numpy array with zeros
  10. How can I check if one two-dimensional NumPy array contains a specific pattern of values inside it?
  11. Finding the consecutive zeros in a numpy array
  12. Numpy ValueError: setting an array element with a sequence. This message may appear without the existing of a sequence?
  13. Numpy: Fix array with rows of different lengths by filling the empty elements with zeros
  14. Numpy – add row to array
  15. Selecting Random Windows from Multidimensional Numpy Array Rows
  16. Find elements of array one nearest to elements of array two
  17. how to perform max/mean pooling on a 2d array using numpy
  18. How to find the index of a value in 2d array in Python?
  19. find and delete from more-dimensional numpy array
  20. Initialise numpy array of unknown length
  21. Why does numpy.linalg.solve() offer more precise matrix inversions than numpy.linalg.inv()?
  22. load csv into 2D matrix with numpy for plotting
  23. numpy division with RuntimeWarning: invalid value encountered in double_scalars
  24. How does the axis parameter from NumPy work?
  25. numpy – evaluate function on a grid of points
  26. numpy arange: how to make “precise” array of floats?
  27. Shift rows of a numpy array independently
  28. Interweaving two NumPy arrays efficiently
  29. Formatting floats in a numpy array [duplicate]
  30. How to convert a 3d numpy array to 2d

Leave a Comment