Selecting rows from a NumPy ndarray,"test[numpy.apply_along_axis(lambda x: x[1] in wanted, 1, test)]"