Python: Elegant and efficient ways to mask a list

by

If you are using numpy:

>>> import numpy as np
>>> a = np.arange(8)
>>> mask = np.array([False, False, False, False, True, True, True, False], dtype=np.bool)
>>> a[mask]
array([4, 5, 6])

If you are not using numpy you are looking for itertools.compress

>>> from itertools import compress
>>> a = range(8)
>>> mask = [False, False, False, False, True, True, True, False]
>>> list(compress(a, mask))
[4, 5, 6]

More Related Contents:

  1. I have a list of file names and I need to be able to count how many of the same file gets repeated for each file name [closed]
  2. How are array types (lists and tuples) and for loops related in Python? [closed]
  3. What are the advantages of NumPy over regular Python lists?
  4. How is Python’s List Implemented?
  5. Python List vs. Array – when to use?
  6. How to read a text file into a list or an array with Python
  7. Flatten list of lists [duplicate]
  8. Can I create a “view” on a Python list?
  9. Find nearest indices for one array against all values in another array – Python / NumPy
  10. Sort multidimensional array based on 2nd element of the subarray
  11. Initializing a list to a known number of elements in Python [duplicate]
  12. How to read an array of integers from single line of input in python3
  13. Implement list-like index access in Python
  14. List to array conversion to use ravel() function
  15. Numpy individual element access slower than for lists
  16. How to group elements in python by n elements [duplicate]
  17. How do I declare an array in Python?
  18. Identify duplicate values in a list in Python
  19. How to access a column in a list of lists in python
  20. How to sort multidimensional array by column?
  21. Python list([]) and []
  22. Python recursive function to display all subsets of given set
  23. Unexpected Behavior of Extend with a list in Python [duplicate]
  24. Optimal method to find the max of sublist items within list
  25. How to group dataframe rows into list in pandas groupby
  26. Modifying a list while iterating over it – why not? [duplicate]
  27. Python 3 range Vs Python 2 range
  28. Appending to 2D lists in Python [duplicate]
  29. Why does creating a list from a list make it larger?
  30. Shift rows of a numpy array independently

Leave a Comment