Phân tầng một thí nghiệm
Bạn đang làm việc với một cơ quan chính phủ muốn thực hiện một thí nghiệm về cách một số chính sách nhất định ảnh hưởng đến tài sản ròng của cá nhân ở nhiều khu vực.
Họ nhờ bạn hỗ trợ thiết kế thí nghiệm. Họ cảnh báo rằng có khả năng tồn tại một nhóm nhỏ người dùng vốn đã có tài sản ròng cao và lo ngại nhóm này có thể lấn át mọi kết quả thí nghiệm quan sát được. Bạn biết phải làm gì rồi!
Hãy vận dụng kiến thức về thiết kế thí nghiệm để thực hiện phân bổ ngẫu nhiên theo khối (block randomization), phân tầng theo cột high_wealth trong DataFrame wealth_data đã cung cấp. DataFrame có 2000 dòng, trong đó có 200 đối tượng có tài sản ròng cao (high_wealth bằng 1).
Bài tập này là một phần của khóa học
Thiết kế thí nghiệm với Python
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# 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'