获取与python中另一列的最大值对应的值

Question

我想知道“stasjon kd”列中“std tid”列的最大值和最小值的相关站点代码，该列按tog-nr分组。我已经编写了以下代码。但它给了我错误。你知道吗

def stasjon_fra_to(path):
    stasjon_kd,std_tid = path
    np.where('std_tid' == arr.max(),'stasjon_kd',NA)

std_1['Stajon'] = std_1[['stasjon_kd','std_tid']].apply(stasjon_fra_to)

错误是：

NameError: ("global name 'arr' is not defined", u'occurred at index stasjon_kd')

虽然我不确定上面的代码是否会引导我找到正确的答案。你知道吗

sample of sorted data for train number 44