Filter Numpy Array with optional argument

Question

I am building a function which should prepare my data depending on the input. The variable x_imp contains indices on which features are important. However sometimes