Convert NumPy array to 0 or 1 based on threshold

by

np.where

np.where(a > 0.5, 1, 0)
# array([0, 0, 0, 1, 1, 1])

Boolean basking with astype

(a > .5).astype(int)
# array([0, 0, 0, 1, 1, 1])

np.select

np.select([a <= .5, a>.5], [np.zeros_like(a), np.ones_like(a)])
# array([ 0.,  0.,  0.,  1.,  1.,  1.])

Special case: np.round

This is the best solution if your array values are floating values between 0 and 1 and your threshold is 0.5.

a.round()
# array([0., 0., 0., 1., 1., 1.])

More Related Contents:

  1. Find the row indexes of several values in a numpy array
  2. How to print the full NumPy array, without truncation?
  3. Convert Python sequence to NumPy array, filling missing values
  4. Numpy array assignment with copy
  5. numpy matrix vector multiplication [duplicate]
  6. Repeating each element of a numpy array 5 times
  7. how does multiplication differ for NumPy Matrix vs Array classes?
  8. Get the position of the largest value in a multi-dimensional NumPy array
  9. Ambiguity in Pandas Dataframe / Numpy Array “axis” definition
  10. How to split/partition a dataset into training and test datasets for, e.g., cross validation?
  11. How to create a numpy array of arbitrary length strings?
  12. Numpy modify array in place?
  13. Subsetting a 2D numpy array
  14. How to normalize a 2-dimensional numpy array in python less verbose?
  15. What is the difference between ndarray and array in NumPy?
  16. Removing elements from an array that are in another array
  17. Creating same random number sequence in Python, NumPy and R
  18. Can numpy bincount work with 2D arrays?
  19. Stocking large numbers into numpy array
  20. check for identical rows in different numpy arrays
  21. pd.Timestamp versus np.datetime64: are they interchangeable for selected uses?
  22. Numpy: For every element in one array, find the index in another array
  23. Prevent numpy from creating a multidimensional array
  24. Sort invariant for numpy.argsort with multiple dimensions
  25. Numpy isnan() fails on an array of floats (from pandas dataframe apply)
  26. Inverting a numpy boolean array using ~
  27. Numpy minimum in (row, column) format
  28. Python: how to store a numpy multidimensional array in PyTables?
  29. get the i-th slice of the k-th dimension in a numpy array
  30. Convert list of lists with different lengths to a numpy array [duplicate]

Leave a Comment