Menu
I have an array of booleans that says which columns of another array I should drop.
For example:
selections = [True, False, True]
data = [[ 1, 2, 3 ],
[ 4, 5, 6 ]]
I would like to have the following one:
new_data = [[ 1, 3 ],
[ 4, 6 ]
All arrays are numpy.array in Python 2.7.
699
In NumPy, you filter an array using a boolean index list. A boolean index list is a list of booleans corresponding to indexes in the array. If the value at an index is True that element is contained in the filtered array, if the value at that index is False that element is excluded from the filtered array.
numpy. array is just a convenience function to create an ndarray ; it is not a class itself. You can also create an array using numpy. ndarray , but it is not the recommended way.
Once you actually use numpy.arrays, it all works:
import numpy as np
selections = np.array([True, False, True])
data = np.array([[ 1, 2, 3 ],
[ 4, 5, 6 ]])
>>> data[:, selections]
array([[1, 3],
[4, 6]])
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With
