pandas: select sub-datasets by complex condition in each group

Question

I need to select a subset from the given dataframe. Here is the df:

import pandas as pd
import numpy as np

df = pd.DataFrame({
                   \'custom_id