Стратификация эксперимента
Вы работаете с государственной организацией, которая хочет провести эксперимент, чтобы оценить, как определённые политические меры влияют на чистое благосостояние граждан в различных регионах.
Организация обратилась к вам за помощью в разработке экспериментального дизайна. Вас предупредили, что среди участников, вероятно, есть небольшая группа людей с высоким уровнем благосостояния, и существует опасение, что она может исказить результаты эксперимента. Вы знаете, как с этим справиться!
Применив знания в области экспериментального дизайна, выполните блочную рандомизацию, проведя стратификацию по столбцу high_wealth в DataFrame wealth_data. DataFrame содержит 2000 строк, из которых 200 относятся к субъектам с высоким уровнем благосостояния (значение high_wealth равно 1).
Это упражнение является частью курса
Планирование экспериментов в Python
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Create the first block
strata_1 = wealth_data[wealth_data[____] == ____]
strata_1['Block'] = ____
# Create two groups assigning to Treatment or Control
strata_1_g1 = strata_1.____(____, replace=False)
strata_1_g1['T_C'] = ____
strata_1_g2 = strata_1.drop(strata_1_g1.index)
strata_1_g2[____] = 'C'