pandas dataframe类型数据通过元素的值获取其所在的行号

今天在处理数据时，要获得某一列的最大值，并且返回该最大值所在的行号

获取某一列的最大值很简单，可以通过

import pandas as pd
# 添加表头
names = ['x1', 'x2', 'x3', 'x4', 'x5', 'x6', 'x7', 'x8', 'x9', ]
#新建dataframe类型数据，a表示数据，是列表类型
df = pd.DataFrame(columns=names, data=a)
# 选择x4列最大的值
x4_max = df['x4'].max()

这样就可以实现找到该列的最大值了，那么怎么返回该值所在的行号呢？
另外，还需注意，最大值可能不止一个，有多个相同的最大值
我们找出这些极值所对应的行，再存到新的dataframe对象里，可以进行其他相应操作
利用.loc属性找到极值所对应的行

rows = pd.DataFrame(columns=names, data=df.loc[df['x4'] == x4_max])
print(df.loc[df['x4'] == x4_max])
# 数据保存
rows.to_csv('first.csv')