Estratificação de um experimento
Você está trabalhando com uma organização governamental que deseja realizar um experimento sobre como algumas políticas governamentais específicas afetam a riqueza líquida dos indivíduos em diversas áreas.
Eles entraram em contato com você para ajudar a montar o projeto experimental. Eles alertaram você de que é provável que haja um pequeno grupo de usuários que já tenham um alto patrimônio líquido e estão preocupados que esse grupo possa ofuscar qualquer resultado experimental observado. Você sabe exatamente o que fazer!
Use seu conhecimento de projeto experimental para realizar a randomização em bloco, estratificando pela coluna high_wealth
no DataFrame wealth_data
fornecido. Há 2.000 linhas no DataFrame com 200 indivíduos com alto patrimônio líquido (high_wealth
é 1).
Este exercício faz parte do curso
Projeto experimental em Python
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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'