How to filter columns in numpy ndarray

Question

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.

Ami Tavory · Accepted Answer · 2016-03-04 18:16:19Z

9

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]])

answered Mar 4, 2016 at 18:16

Ami Tavory

76.7k13 gold badges152 silver badges196 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to filter columns in numpy ndarray

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related