我正在嘗試學習如何在 python 3.7 中執行一些我通常在 Stata 中輕松執行的操作。我正在處理這樣的數據框：estimation_window    group_id         y            x   0                       1             3.17         231                       1             4.76         261                       2             4.17         731                       2             8.70         720                       2             11.45        16我想要一個 for 循環來估計 group_id 中每個不同組的線性回歸（y on x），只使用虛擬 estimation_window 等于 1 的觀察值。然后，我想將三列添加回我的初始數據框：一列包含預測值，一列包含每個回歸的 beta（x 系數），一列包含每個回歸的截距。應該為所有觀察值計算這些值，而不僅僅是虛擬 estimation_window 等于 1 的那些值。生成的數據集應如下所示：estimation_window    group_id         y            x        predicted_val    beta_coeff  alpha_coeff0                       1             3.17         23           3.10            0.32         1.43        1                       1             4.76         26           4.00            0.32         1.431                       2             4.17         73           4.10            0.75         0.951                       2             8.70         72           8.50            0.75         0.950                       2             11.45        16           10.30           0.75         0.95我嘗試使用 statsmodels 但無法弄清楚。