Estratificar un experimento
Estás trabajando con una organización gubernamental que quiere llevar a cabo un experimento sobre cómo afectan algunas políticas gubernamentales concretas a la riqueza neta de los individuos en una serie de áreas.
Se han puesto en contacto contigo para que les ayudes a establecer el diseño experimental. Te han advertido de que es probable que haya un pequeño grupo de usuarios que ya tienen un patrimonio neto elevado y les preocupa que este grupo pueda eclipsar cualquier resultado experimental que se observe. ¡Ya sabes lo que hay que hacer!
Utiliza tus conocimientos de diseño experimental para llevar a cabo la aleatorización por bloques, estratificando por la columna high_wealth
del DataFrame proporcionado wealth_data
. Hay 2000 filas en el DataFrame con 200 sujetos de alto patrimonio neto (high_wealth
es 1).
Este ejercicio forma parte del curso
Diseño experimental en Python
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# 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'