Julia DataFrames,在特定索引处插入新行

时间:2018-07-24 18:05:28

标签: dataframe insert julia concat

是否可以在特定索引处向现有数据帧添加行?

例如您有一个3行1列的数据框

df = DataFrame(x = [2,3,4])

X
2
3
4

执行以下操作的任何方式:

insert!(df, 1, [1])

为了获得

X
1
2
3
4

我知道我可以合并两个数据帧df = [df1; df2],但我希望每当要插入一行时都避免对大型DF进行垃圾处理。

1 个答案:

答案 0 :(得分:2)

我想您想就地进行。然后,您可以像这样使用insert!函数:

julia> df = DataFrame(x = [1,2,3], y = ["a", "b", "c"])
3×2 DataFrames.DataFrame
│ Row │ x │ y │
├─────┼───┼───┤
│ 1   │ 1 │ a │
│ 2   │ 2 │ b │
│ 3   │ 3 │ c │

julia> foreach((v,n) -> insert!(df[n], 2, v), [4, "d"], names(df))

julia> df
4×2 DataFrames.DataFrame
│ Row │ x │ y │
├─────┼───┼───┤
│ 1   │ 1 │ a │
│ 2   │ 4 │ d │
│ 3   │ 2 │ b │
│ 4   │ 3 │ c │

当然,您必须确保添加的集合中的列数正确。

如果您接受使用DataFrame的未导出内部结构,则可以做得更简单:

julia> df = DataFrame(x = [1,2,3], y = ["a", "b", "c"])
3×2 DataFrames.DataFrame
│ Row │ x │ y │
├─────┼───┼───┤
│ 1   │ 1 │ a │
│ 2   │ 2 │ b │
│ 3   │ 3 │ c │

julia> insert!.(df.columns, 2, [4, "d"])
2-element Array{Array{T,1} where T,1}:
 [1, 4, 2, 3]
 String["a", "d", "b", "c"]

julia> df
4×2 DataFrames.DataFrame
│ Row │ x │ y │
├─────┼───┼───┤
│ 1   │ 1 │ a │
│ 2   │ 4 │ d │
│ 3   │ 2 │ b │
│ 4   │ 3 │ c │

更新

由于Julia 1.0 df.columns不起作用。而是写:

insert!.(eachcol(df, false), 2, [4, "d"])
相关问题