np.isin(data[:, 0], match)
Example:
data = np.array([[1, 4],[5, 2],[2, 4]]) match = np.array([2, 4]) np.isin(data[:,0], match) array([False, False, True]) data[np.isin(data[:, 0], match)] array([[2, 4]])
np.isin(data[:, 0], match)
Example:
data = np.array([[1, 4],[5, 2],[2, 4]]) match = np.array([2, 4]) np.isin(data[:,0], match) array([False, False, True]) data[np.isin(data[:, 0], match)] array([[2, 4]])