首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >julia dataframe -如何在下一行中逐组获得值

julia dataframe -如何在下一行中逐组获得值
EN

Stack Overflow用户
提问于 2021-03-16 07:23:52
回答 1查看 169关注 0票数 2

在朱莉娅DataFrame中,我如何通过和使用下一行的值来处理组?

例如:

代码语言:javascript
运行
复制
using DataFrames, DataFramesMeta
df = DataFrame(grp=["one", "one", "two", "two", "three"], val=[1, 2, 3, 4, 5])
# Row │ grp     val   
#     │ String  Int64 
#─────┼───────────────
#   1 │ one         1
#   2 │ one         2
#   3 │ two         3
#   4 │ two         4
#   5 │ three       5
   
@combine(groupby(df, :grp),
               count = length(:val),
               first_val = first(:val),
               #next_val = next(:val)
             )
#3×3 DataFrame
# Row │ grp     count  first_val 
#     │ String  Int64  Int64     
#─────┼──────────────────────────
#   1 │ one         2          1
#   2 │ two         2          3
#   3 │ three       1          5

# I would like to obtain:

# Row │ grp     count  first_val next_val
#     │ String  Int64  Int64     
#─────┼──────────────────────────
#   1 │ one         2          1 2
#   2 │ two         2          3 4
#   3 │ three       1          5 NA
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-03-16 09:08:38

对于朱莉娅DataFrames.jl,它应该是:

代码语言:javascript
运行
复制
julia> combine(groupby(df, :grp),
               nrow => :count,
               :val => first => :first_val,
               :val => (x -> length(x) > 1 ? x[2] : missing) => :next_val)
3×4 DataFrame
 Row │ grp     count  first_val  next_val
     │ String  Int64  Int64      Int64?
─────┼────────────────────────────────────
   1 │ one         2          1         2
   2 │ two         2          3         4
   3 │ three       1          5   missing

如果您接受额外的包,那么对于ShiftedArrays.jl,它将是:

代码语言:javascript
运行
复制
julia> using ShiftedArrays

julia> combine(groupby(df, :grp),
               nrow => :count,
               :val => first => :first_val,
               :val => first∘lead => :next_val)
3×4 DataFrame
 Row │ grp     count  first_val  next_val
     │ String  Int64  Int64      Int64?
─────┼────────────────────────────────────
   1 │ one         2          1         2
   2 │ two         2          3         4
   3 │ three       1          5   missing

这里的内容与自动生成的列名相同:

代码语言:javascript
运行
复制
julia> combine(groupby(df, :grp), nrow, :val => first, :val => first∘lead)
3×4 DataFrame
 Row │ grp     nrow   val_first  val_first_lead
     │ String  Int64  Int64      Int64?
─────┼──────────────────────────────────────────
   1 │ one         2          1               2
   2 │ two         2          3               4
   3 │ three       1          5         missing
票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/66650628

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档