結果 : subset dataframe in python based on condition