Add a conditional column in Python | Google SQL Question