python - Julia 将分区数据帧存储到字典中

Question

尝试将 python 脚本翻译成 Julia，而 Julia 似乎不止于此（诚然，我对 Julia 的理解非常原始，所以我预计会遇到一些困难）。简而言之，我试图通过列向量（具有 32 个级别！！）拆分数据帧，然后将这些分区数据帧写入文本。在 python 中，我被亲切地建议写这样的东西来分割数据帧并将它们存储在一个字典中：

injuries = {injury: df[df['Type'] == injury] for injury in df['Type'].unique()}

injuries['BROKEN PELVIS']

有没有人知道如何在 Julia 中实现类似的目标？我想 Julia 在语法上是相似的，但到目前为止我所有的努力都没有结果。真诚感谢任何输入。谢谢。大通CB

Answer 1

让我们弥补一些数据：

df = DataFrame(val=rand(8), injury = [rep("shoulder",4), rep("leg",4)])

此版本与 Python 版本相同：

injuries = [injury=>df[df[:injury] .== injury,:] for injury in unique(df[:injury])]

现在injuries是一个 DataFrame 字典，每个伤害一个 DataFrame。然后你可以这样做：

injuries["shoulder"]

它返回一个只有肩部受伤的 DataFrame。