encontrar índice correspondiente a pandas de valor máximo
my_max_ind = data['x1'].idxmax() # Index of maximum in column
print(my_max_ind)
# 4
Confused Cobra