Find The Index Of Maximum Element In Python