Modifying a subset of rows in a pandas dataframe #shorts