dataframe - Julia DataFrame 用 LOCF 填充 NA

Question

有没有什么快速的方法可以将 DataFrame 的 NA 值转换为最后观察到的值？

using DataFrames

d = @data [1,NA,5,NA,NA]
df = DataFrame(d=d)

result = filled_with_locf(df)

expected = [1,1,5,5,5]

LOCF = 最后一次观察结转

Answer 1

扩展注释 oneliner，如果我们定义 locf作为:

locf(v) = v[cummax([i*!isna(v[i]) for i=1:length(v)])]

然后，

nona_df = DataFrame(Any[locf(df[c]) for c in names(df)],names(df))

和，

julia> nona_df
5×1 DataFrames.DataFrame
│ Row │ d │
├─────┼───┤
│ 1   │ 1 │
│ 2   │ 1 │
│ 3   │ 5 │
│ 4   │ 5 │
│ 5   │ 5 │