How do I use numpy.where()? What should I pass, and what does the result mean? [closed]

by

After fiddling around for a while, I figured things out, and am posting them here hoping it will help others.

Intuitively, np.where is like asking “tell me where in this array, entries satisfy a given condition“.

>>> a = np.arange(5,10)
>>> np.where(a < 8)       # tell me where in a, entries are < 8
(array([0, 1, 2]),)       # answer: entries indexed by 0, 1, 2

It can also be used to get entries in array that satisfy the condition:

>>> a[np.where(a < 8)] 
array([5, 6, 7])          # selects from a entries 0, 1, 2

When a is a 2d array, np.where() returns an array of row idx’s, and an array of col idx’s:

>>> a = np.arange(4,10).reshape(2,3)
array([[4, 5, 6],
       [7, 8, 9]])
>>> np.where(a > 8)
(array(1), array(2))

As in the 1d case, we can use np.where() to get entries in the 2d array that satisfy the condition:

>>> a[np.where(a > 8)] # selects from a entries 0, 1, 2

array([9])

Note, when a is 1d, np.where() still returns an array of row idx’s and an array of col idx’s, but columns are of length 1, so latter is empty array.

More Related Contents:

  1. How do I read CSV data into a record array in NumPy?
  2. Working with big data in python and numpy, not enough ram, how to save partial results on disc?
  3. Python out of memory on large CSV file (numpy)
  4. python numpy/scipy curve fitting
  5. Multiple linear regression in Python
  6. matplotlib – extracting data from contour lines
  7. Fitting a Normal distribution to 1D data
  8. Populate a Pandas SparseDataFrame from a SciPy Sparse Matrix
  9. Installing NumPy and SciPy on 64-bit Windows (with Pip)
  10. How to extend an array in-place in Numpy?
  11. Interpolation over an irregular grid
  12. fitting exponential decay with no initial guessing
  13. Overflow Error in Python’s numpy.exp function
  14. How to plot line (polygonal chain) with numpy/scipy/matplotlib with minimal smoothing
  15. Storing numpy sparse matrix in HDF5 (PyTables)
  16. How can I efficiently process a numpy array in blocks similar to Matlab’s blkproc (blockproc) function
  17. Creating a Confidence Ellipsis in a scatterplot using matplotlib
  18. Load just part of an image in python
  19. generalized cumulative functions in NumPy/SciPy?
  20. Vectorized spatial distance in python using numpy
  21. Principal component analysis in Python
  22. Parse a Pandas column to Datetime when importing table from SQL database and filtering rows by date
  23. Python 4D linear interpolation on a rectangular grid
  24. Consecutive, Overlapping Subsets of Array (NumPy, Python)
  25. Interpolating a closed curve using scipy
  26. Procrustes Analysis with NumPy?
  27. How to include constraint to Scipy NNLS function solution so that it sums to 1
  28. scipy convolve2d outputs wrong values
  29. Computing cross-correlation function?
  30. Understanding scipy deconvolve

Leave a Comment