dataframe - Julia DataFrame 列名称

Question

我想命名空白DataFrame的列名。

我在 v0.6 中(重新)命名 DataFrame 的多个列时遇到问题。

我尝试过生成名称:

df = DataFrame()
nms = [":x$i" for i in 1:2]
df[nms[1] = rand(10)]
df[nms[2] = rand(10)]

但是名称的符号要求不适用于 $ 宏。有人解决了吗？

Answer 1

df = DataFrame()
nms = [Symbol("x$i") for i in 1:2]
df[nms[1]] = rand(10)
df[nms[2]] = rand(10)

可以工作并且与相关代码最相似。关键是使用 Julia 符号而不是字符串来索引 DataFrame。

创建 df 的更紧凑方法是使用 DataFrame 构造函数，该构造函数将列作为参数。有一个，但是参数是命名参数，所以你需要splat(即...)一个Dict来插入参数。 Dict 本身需要以编程方式生成。要维护 Dict 中列的特定顺序，您需要一个 OrderedDict。结果是:

using DataFrames, DataStructures

df = DataFrame(;OrderedDict((Symbol("x$i")=>rand(10) for i=1:3)...)...)