Generate a column based on a constraint in pandas Generate a column based on a constraint in pandas numpy numpy