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

尝试将 python 脚本翻译成 Julia 和 Julia 似乎更胜一筹(不可否认，我对 Julia 的理解非常原始，所以我预计会有一些困难)。简而言之，我试图通过列向量(有 32 个级别!!)分割数据帧，然后将这些分区的数据帧写入文本。在 python 中，我被慷慨地建议写这样的东西来分割数据帧并将它们存储在一个字典中:

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

injuries['BROKEN PELVIS']

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

最佳答案

让我们组成一些数据:

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"]

它返回一个只有肩部受伤的数据帧。