从给定的 Dataframe 返回排序的 Dataframe

Question

我需要编写一个函数，它接受一个名为 theData 的数据框、一个名为 groupBy 的列和另一个名为 orderBy 的列，并创建另一个表示数据透视表的排序数据框 table。

需要分组'groupBy'，使用orderBy的平均值从大到小排序

现在，有些事情我无法改变...

theData 是我正在解释的数据场 groupBy 是分组依据的列 orderBy 是要按平均数从大到小排序的列。

我也得到了以下格式...

def createPivot(theData, groupBy, orderBy):
# YOUR CODE HERE
return retDF

到目前为止，这就是我所拥有的...

def createPivot(theData, groupBy, orderBy):
retDF = theData.groupby(groupBy).mean().reset_index().sort_values(by=orderBy, inplace=True)
return retDF

我从测试代码中收到一条错误消息，告诉我我没有返回数据帧。我的教授也告诉我，我应该基本上可以用一行代码来完成任务。我想我想太多了，所以一个简单的答案在这里会非常有帮助！

谢谢！

Answer 1

根据你的描述，这是我想出来的：

def createPivot(theData, groupBy, orderBy):
    retDF = theData.groupby(groupBy).mean().sort_values(by=orderBy,ascending=False)
    return retDF

Returning a sorted Dataframe from a given Dataframe