Replacing Numpy elements if condition is met

by 
>>> import numpy as np
>>> a = np.random.randint(0, 5, size=(5, 4))
>>> a
array([[4, 2, 1, 1],
       [3, 0, 1, 2],
       [2, 0, 1, 1],
       [4, 0, 2, 3],
       [0, 0, 0, 2]])
>>> b = a < 3
>>> b
array([[False,  True,  True,  True],
       [False,  True,  True,  True],
       [ True,  True,  True,  True],
       [False,  True,  True, False],
       [ True,  True,  True,  True]], dtype=bool)
>>> 
>>> c = b.astype(int)
>>> c
array([[0, 1, 1, 1],
       [0, 1, 1, 1],
       [1, 1, 1, 1],
       [0, 1, 1, 0],
       [1, 1, 1, 1]])

You can shorten this with:

>>> c = (a < 3).astype(int)

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. Is there any numpy group by function?
  5. Computing the correlation coefficient between two multi-dimensional arrays
  6. Removing nan values from an array
  7. Why is numpy’s einsum faster than numpy’s built in functions?
  8. How to return 0 with divide by zero
  9. Efficiently return the index of the first value satisfying condition in array
  10. How to save and load numpy.array() data properly?
  11. How do I access the ith column of a NumPy multidimensional array?
  12. Interweaving two numpy arrays
  13. python how to pad numpy array with zeros
  14. How can I check if one two-dimensional NumPy array contains a specific pattern of values inside it?
  15. Finding the consecutive zeros in a numpy array
  16. Numpy ValueError: setting an array element with a sequence. This message may appear without the existing of a sequence?
  17. Numpy: Fix array with rows of different lengths by filling the empty elements with zeros
  18. Numpy – add row to array
  19. How to split/partition a dataset into training and test datasets for, e.g., cross validation?
  20. Selecting Random Windows from Multidimensional Numpy Array Rows
  21. Find elements of array one nearest to elements of array two
  22. how to perform max/mean pooling on a 2d array using numpy
  23. How to find the index of a value in 2d array in Python?
  24. find and delete from more-dimensional numpy array
  25. Initialise numpy array of unknown length
  26. Creating same random number sequence in Python, NumPy and R
  27. Why does numpy.linalg.solve() offer more precise matrix inversions than numpy.linalg.inv()?
  28. load csv into 2D matrix with numpy for plotting
  29. Efficient dot products of large memory-mapped arrays
  30. Mapping a NumPy array in place

Leave a Comment