Stratifiera ett experiment
Du arbetar med en statlig organisation som vill genomföra ett experiment kring hur vissa regeringspolicyer påverkar individers nettförmögenhet i ett antal områden.
De har vänt sig till dig för hjälp med att utforma experimentet. De har varnat dig för att det troligen finns en liten grupp deltagare som redan har hög nettoförmögenhet, och de är oroliga att denna grupp kan överskugga de experimentella resultaten. Du vet precis vad du ska göra!
Använd dina kunskaper i experimentdesign för att genomföra blockrandomisering och stratifiera utifrån kolumnen high_wealth i den tillhandahållna DataFrame:en wealth_data. DataFrame:en innehåller 2 000 rader med 200 deltagare med hög nettoförmögenhet (high_wealth är 1).
Den här övningen är en del av kursen
Experimentell design i Python
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# 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'