# Before df['order_flag'] = df.apply( lambda row: 'high' if row['net_revenue'] > 50000 else 'low', axis=1 ) # After df['order_flag'] = np.where(df['net_revenue'] > 50000, 'high', 'low')