Stratyfikacja eksperymentu
Pracujesz z organizacją rządową, która chce przeprowadzić eksperyment dotyczący wpływu wybranych polityk rządowych na zamożność netto osób w kilku obszarach.
Zwrócono się do ciebie z prośbą o pomoc w zaprojektowaniu eksperymentu. Ostrzeżono cię, że w danych może znajdować się niewielka grupa osób o już wysokiej zamożności netto i że ta grupa mogłaby zaburzyć obserwowane wyniki eksperymentu. Wiesz dokładnie, co zrobić!
Wykorzystaj swoją wiedzę z zakresu projektowania eksperymentów i przeprowadź blokową randomizację, stosując stratyfikację według kolumny high_wealth w dostarczonym DataFrame wealth_data. DataFrame zawiera 2000 wierszy, z czego 200 to osoby o wysokiej zamożności netto (high_wealth równe 1).
To ćwiczenie jest częścią kursu
Projektowanie eksperymentów w Pythonie
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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'