julia - 来自 Julia 中 Dataframe 列的向量

Question

我有一个DataFrame

df = DataFrame(x = 1:3, y = 4:6)

3×2 DataFrame
 Row │ x      y     
     │ Int64  Int64 
─────┼──────────────
   1 │     1      4
   2 │     2      5
   3 │     3      6

如何将其中一列提取为 Vector？

我知道我可以做 df[:,:x] 或 df.x，但是有没有办法用函数来做呢？我问的原因是我使用 Chain.jl 包并想做类似的事情

@chain df begin
    some_manipulations_here
    pull(:x)
end

Answer 1

您可以执行以下操作之一:

julia> df = DataFrame(x = 1:3, y = 4:6)
3×2 DataFrame
 Row │ x      y
     │ Int64  Int64
─────┼──────────────
   1 │     1      4
   2 │     2      5
   3 │     3      6

julia> @chain df begin
       _.x
       end
3-element Vector{Int64}:
 1
 2
 3

julia> @chain df begin
       getproperty(:x) # same as above
       end
3-element Vector{Int64}:
 1
 2
 3

julia> @chain df begin
       getindex(!, :x) # also _[!, :x]
       end
3-element Vector{Int64}:
 1
 2
 3

julia> @chain df begin
       getindex(:, :x) # also _[:, :x]
       end
3-element Vector{Int64}:
 1
 2
 3

可能是第一个选项(使用 _.x 在实践中是最简单的。

我展示了其他选项以强调所有特殊语法，如 df.x 或 df[!, :x] 实际上是函数调用，特殊语法只是为了方便。