Типова задача в data science — розбити датафрейм за певною групою, застосувати перетворення до кожної групи, а потім зібрати все назад в один датафрейм. У R цей підхід настільки поширений, що має назву split-apply-combine. У курсі Intermediate R for Finance ви розглянете низку таких задач і корисних функцій для їх розв'язання, а зараз виконаємо простий приклад.

Припустімо, що для датафрейму cash ви хочете подвоїти cash_flow для компанії A та потроїти для компанії B:

grouping <- cash$company
split_cash <- split(cash, grouping)

# Доступ до стовпця cash_flow кожного елемента списку:
split_cash$A$cash_flow
[1] 1000 4000  550

split_cash$A$cash_flow <- split_cash$A$cash_flow * 2
split_cash$B$cash_flow <- split_cash$B$cash_flow * 3

new_cash <- unsplit(split_cash, grouping)

Зверніть увагу, як саме відбувається доступ до стовпця cash_flow. Перший $ звертається до елемента A списку split_cash. Другий $ звертається до стовпця cash_flow датафрейму в A.

Вітаємо!

Create Your Free Account