Select elements of numpy array via boolean mask array

by

You probably want something like this:

>>> a = np.array([True, True, True, False, False])
>>> b = np.array([[1,2,3,4,5], [1,2,3,4,5]])
>>> b[:,a]
array([[1, 2, 3],
       [1, 2, 3]])

Note that for this kind of indexing to work, it needs to be an ndarray, like you were using, not a list, or it’ll interpret the False and True as 0 and 1 and give you those columns:

>>> b[:,[True, True, True, False, False]]   
array([[2, 2, 2, 1, 1],
       [2, 2, 2, 1, 1]])

More Related Contents:

  1. How to apply a disc shaped mask to a NumPy array?
  2. using an numpy array as indices of the 2nd dim of another array? [duplicate]
  3. How can I map True/False to 1/0 in a Pandas DataFrame?
  4. Update row values where certain condition is met in pandas
  5. How to extract an arbitrary line of values from a numpy array?
  6. Comparing previous row values in Pandas DataFrame
  7. Get the position of the largest value in a multi-dimensional NumPy array
  8. Why does indexing numpy arrays with brackets and commas differ in behavior?
  9. Increment Numpy array with repeated indices
  10. Generalise slicing operation in a NumPy array
  11. Find index where elements change value numpy
  12. Python Bool and int comparison and indexing on list with boolean values
  13. Numpy: For every element in one array, find the index in another array
  14. Index all *except* one item in python
  15. Numpy sub-array assignment with advanced, mixed indexing
  16. python tilde unary operator as negation numpy bool array
  17. Filtering pandas dataframe with multiple Boolean columns
  18. How is numpy’s fancy indexing implemented?
  19. Access n-th dimension in python [duplicate]
  20. numpy array TypeError: only integer scalar arrays can be converted to a scalar index
  21. Cython: (Why / When) Is it preferable to use Py_ssize_t for indexing?
  22. Get the indices of N highest values in an ndarray
  23. Rank items in an array using Python/NumPy, without sorting array twice
  24. Numpy argsort – what is it doing?
  25. Subsetting a 2D numpy array
  26. How to normalize a 2-dimensional numpy array in python less verbose?
  27. How to get a normal distribution within a range in numpy? [duplicate]
  28. Annotate Time Series plot in Matplotlib
  29. Index Error: list index out of range (Python) [duplicate]
  30. 3d Numpy array to 2d

Leave a Comment