Python Pandas Dataframe challenge how do I avoid Iterrows for this scenario