将项目从Python转Golang:Gota如何实现类似Pandas的nlargest取前20行?
在Gota中实现类似Pandas nlargest的前N行选取操作
你可以直接使用Gota DataFrame的Head()方法提取排序后的前20行,结合你已有的排序代码,完整实现如下:
// 按Change列降序排序 sorted := valuesDf.Arrange(dataframe.RevSort("Change")) // 选取前20行数据 top20 := sorted.Head(20) // 计算Change列的平均值 meanValue, err := top20.Col("Change").Mean() if err != nil { // 此处添加错误处理逻辑,比如打印错误或返回错误 }
说明
Head(n int)方法会返回DataFrame的前n行数据,如果原DataFrame的行数小于n,则返回全部行,行为和Pandas的head(n)完全一致。- 结合你之前的
RevSort("Change")降序排序,这个组合操作就等价于Pandas中df.nlargest(20, ['Change'])的功能。
内容的提问来源于stack exchange,提问作者Barry Marples
相关产品推荐
相关产品推荐

