Стратифікація експерименту
Ви співпрацюєте з урядовою організацією, яка хоче провести експеримент про те, як певні державні політики впливають на чисте багатство людей у низці регіонів.
Вони звернулися до вас по допомогу з налаштуванням плану експерименту. Вони попередили, що, ймовірно, є невелика група користувачів із уже високим чистим багатством і є занепокоєння, що ця група може «перекрити» будь-який спостережуваний результат експерименту. Ви знаєте, що робити!
Скористайтеся знаннями з планування експериментів, щоб виконати блочну рандомізацію, стратифікувавши за стовпцем high_wealth у наданому датафреймі wealth_data. У датафреймі 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'