dataframe - 如何计算新列并添加到 Julia 中的数据框？

Question

(Julia 新手)

我正在尝试运行此操作。这是一个最小的工作示例:

df = DataFrame(A = 1:4)

Row A
Int64
1   1
2   2
3   3
4   4

只是一个具有四个值的数据框，1-4。我想添加一个新列，其中每个值等于元素加上之前的元素。换句话说，我想要:

Row A Row B
Int64 Int64
1   1   1
2   2   3
3   3   6
4   4   10

我该怎么做？

我可以编写一个函数来计算所需的数字:

function first(j)
    val = 0
    while j != 0
        val += df.A[j]
        j -= 1
    end
    return val
end

这里的j是元素的索引。 This question还给出了如何在计算后添加列。但是，我不知道如何将这些值变成一个新列。我怀疑应该有比计算数字更简单的方法，用它形成一个列，然后将它也添加到数据框中。

Answer 1

julia> df.B = cumsum(df.A);

julia> df
4×2 DataFrame
 Row │ A      B
     │ Int64  Int64
─────┼──────────────
   1 │     1      1
   2 │     2      3
   3 │     3      6
   4 │     4     10