A/B Testing en Python
Moe Lotfy, PhD
Principal Data Science Manager

Unité d'analyse :
Unité de randomisation :

print(checkout.groupby('checkout_page')[['order_value','purchased']].agg({'sum','count','mean'}))
order_value purchased
mean sum count mean sum count
checkout_page
A 24.956437 61417.791564 2461 0.820333 2461.0 3000
B 29.876202 75915.430125 2541 0.847000 2541.0 3000
C 34.917589 90890.484142 2603 0.867667 2603.0 3000
checkout.groupby('checkout_page')['order_value'].sum()/
checkout.groupby('checkout_page')['purchased'].count()
checkout_page
A 20.472597
B 25.305143
C 30.296828
dtype: float64

# Variance d'une mesure en ratio par la méthode delta
def var_delta(x,y):
x_bar = np.mean(x)
y_bar = np.mean(y)
x_var = np.var(x,ddof=1)
y_var = np.var(y,ddof=1)
cov_xy = np.cov(x,y,ddof=1)[0][1]
# Remarquez qu'on divise par len(x) ici, car le dénominateur de la statistique de test est l'erreur type (=sqrt(var/n))
var_ratio = (x_var/y_bar**2 + y_var*(x_bar**2/y_bar**4) - 2*cov_xy*(x_bar/y_bar**3))/len(x)
return var_ratio
# Calcul du z-test par la méthode delta
ztest_delta(x_control,y_control,x_treatment,y_treatment, alpha = 0.05)
Arguments d'entrée :
x_control : colonne numérateur du ratio par utilisateur pour la variante témoiny_control : colonne dénominateur du ratio par utilisateur pour la variante témoinx_treatment : colonne numérateur du ratio par utilisateur pour la variante testy_treatment : colonne dénominateur du ratio par utilisateur pour la variante testalpha : seuil de signification.Sortie :
mean_control : moyenne de la mesure en ratio du groupe témoinmean_treatment : moyenne de la mesure en ratio du groupe testdifference : écart entre les moyennes test et témoindiff_CI : intervalle de confiance de la différence de moyennesp-value : valeur p bilatérale du z-test# Créer des DataFrame pour les mesures par utilisateur des variantes A et B
A_per_user = pd.DataFrame({'order_value':checkout[checkout['checkout_page']=='A'].groupby('user_id')['order_value'].sum()
,'page_view':checkout[checkout['checkout_page']=='A'].groupby('user_id')['user_id'].count()})
B_per_user = pd.DataFrame({'order_value':checkout[checkout['checkout_page']=='B'].groupby('user_id')['order_value'].sum()
,'page_view':checkout[checkout['checkout_page']=='B'].groupby('user_id')['user_id'].count()})
# Attribuer les colonnes de ratio témoin et test
x_control = A_per_user['order_value']
y_control = A_per_user['page_view']
x_treatment = B_per_user['order_value']
y_treatment = B_per_user['page_view']
# Exécuter un z-test pour des mesures en ratio
ztest_delta(x_control,y_control,x_treatment,y_treatment)
{'mean_control': 20.472597188012,
'mean_treatment': 25.30514337484097,
'difference': 4.833,
'diff_CI': '[4.257, 5.408]',
'p-value': 5.954978880467735e-61}
A/B Testing en Python