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。