Dataframe Indexing In Python Definition Function Math