Джулия сохраняет секционированный фрейм данных в словарь

Попытка перевести скрипт python на Julia и Julia кажется более чем подходящей (правда, у меня очень примитивное понимание Julia, поэтому я ожидал некоторых трудностей). Короче говоря, я пытаюсь разделить фрейм данных на вектор-столбец (который имеет 32 уровня!!), а затем записать эти разделенные фреймы данных в текст. В python мне любезно посоветовали написать что-то вроде этого, чтобы разделить фреймы данных и сохранить их в dict:

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

injuries['BROKEN PELVIS']

Кто-нибудь знает, как добиться чего-то подобного в Джулии? Я бы предположил, что Джулия синтаксически похожа, но все мои усилия до сих пор были бесплодными. Любой вклад искренне ценится. Спасибо. Чейз КБ


python dictionary dataframe julia dictionary-comprehension
person Chase CB    schedule 24.02.2014    source источник

Ответы (1)


arrow_upward
3
arrow_downward

Давайте составим некоторые данные:

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 — это словарь DataFrames, по одному DataFrame для каждой травмы. Затем вы можете просто сделать:

injuries["shoulder"]

и он возвращает DataFrame только с травмами плеча.

person Mr Alpha    schedule 24.02.2014