La solicitud

De Jordan Reyes, Director de Marketing

“La prueba del banner de envío gratis duró todo noviembre y diciembre. Antes de que me digas si funcionó, nuestro responsable de datos dice que primero hay que revisar la división. ¿Puedes confirmar que los dos grupos tienen el tamaño que deberían?”

Tu objetivo

Con test (sesiones del 2025-11-01 al 2025-12-31, con arm = "A" o "B"), asigna a result un diccionario con "A" y "B", el número de sesiones de cada grupo (int), "chi2", el estadístico chi-cuadrado frente a una división 50/50 (redondeado a 3), y "p_value", su valor p con 1 grado de libertad, 2 * (1 - NormalDist().cdf(sqrt(chi2))) (redondeado a 4).

  • Usa el DataFrame web_sessionspendiente
  • Guarda un dict en resultpendiente

Tu código

Python

Cargando el editor…

    ResultadoVista previa de web_sessions · presiona Ejecutar para ver el tuyo
    Vista previa de web_sessions
    session_idcustomer_idstarted_atdevicesource
    1NaN2023-01-01 23:02:24mobilepaid
    2182023-01-01 15:07:49desktoppaid
    3NaN2023-01-01 16:55:46desktopdirect
    4NaN2023-01-01 12:43:28mobileorganic

    Mote revisa tu resultado

    Presiona Ejecutar. Cada fila que cumple el objetivo se vuelve una luciérnaga en tu linterna.

    DataFrames

    Cargados como DataFrames de pandas, junto con pd y np.

    web_sessions40,000 filas · 7 columnas
    • session_idinteger
    • customer_idinteger
    • started_attimestamp
    • devicevarchar
    • sourcevarchar
    • pages_viewedinteger
    • convertedboolean
    Primeras 5 filas de web_sessions
    session_idcustomer_idstarted_atdevicesourcepages_viewedconverted
    1NaN2023-01-01 23:02:24mobilepaid1false
    2182023-01-01 15:07:49desktoppaid1false
    3NaN2023-01-01 16:55:46desktopdirect7false
    4NaN2023-01-01 12:43:28mobileorganic3false
    5NaN2023-01-01 13:06:21mobilesocial2false
    customers2,018 filas · 9 columnas
    • customer_idinteger
    • first_namevarchar
    • last_namevarchar
    • emailvarchar
    • cityvarchar
    • statevarchar
    • signup_datedate
    • segmentvarchar
    • referral_sourcevarchar
    Primeras 5 filas de customers
    customer_idfirst_namelast_nameemailcitystatesignup_datesegmentreferral_source
    1OwenHudsonowen.hudson@icloud.comSan FranciscoCA2022-06-02consumersearch
    2DouglasMendozadmendoza69@gmail.comAtlantaGA2022-06-02consumerNaN
    3HenryBrookshbrooks@lakesidecafe.comLos AngelesCA2022-06-04businessNaN
    4BrianGrahambgraham@bluecedarhomes.comSeattleWA2022-06-06businesspinterest
    5KyleJimenezkjimenez27@yahoo.comIndianapolisIN2022-06-07consumerNaN
    products120 filas · 8 columnas
    • product_idinteger
    • namevarchar
    • categoryvarchar
    • subcategoryvarchar
    • unit_pricedecimal
    • unit_costdecimal
    • launched_ondate
    • is_activeboolean
    Primeras 5 filas de products
    product_idnamecategorysubcategoryunit_priceunit_costlaunched_onis_active
    101Vale Velvet LoveseatFurnitureSofas1659.00NaN2023-09-11true
    102Alder Oak Coffee TableFurnitureTables919.00503.502024-05-16true
    103Orchard Rattan Dining ChairFurnitureChairs789.00429.872023-05-27true
    104Upland Oak BookcaseFurnitureStorage429.00214.702022-12-20true
    105Avery Leather Bed FrameFurnitureBeds449.00232.702021-08-31true
    orders12,000 filas · 7 columnas
    • order_idinteger
    • customer_idinteger
    • order_datetimestamp
    • statusvarchar
    • channelvarchar
    • shipping_feedecimal
    • coupon_codevarchar
    Primeras 5 filas de orders
    order_idcustomer_idorder_datestatuschannelshipping_feecoupon_code
    10011252023-01-01 20:51:19completedweb0.00NaN
    10021562023-01-02 07:58:46completedweb0.00NaN
    1003222023-01-02 10:25:27completedmarketplace4.99WELCOME10
    1004482023-01-02 11:21:51completedmarketplace4.99WELCOME10
    1005822023-01-02 11:43:36completedweb0.00WELCOME10
    order_items30,338 filas · 6 columnas
    • order_item_idinteger
    • order_idinteger
    • product_idinteger
    • quantityinteger
    • unit_pricedecimal
    • discount_pctdecimal
    Primeras 5 filas de order_items
    order_item_idorder_idproduct_idquantityunit_pricediscount_pct
    11001121931.990.00
    21001154311.990.00
    31002154411.990.00
    410021941149.000.00
    51002193219.990.00
    returns1,673 filas · 5 columnas
    • return_idinteger
    • order_item_idinteger
    • returned_ondate
    • reasonvarchar
    • refund_amountdecimal
    Primeras 5 filas de returns
    return_idorder_item_idreturned_onreasonrefund_amount
    1142023-01-10damaged83.69
    2422023-01-10wrong_item19.99
    31052023-01-14wrong_item10.99
    4242023-01-16changed_mind26.99
    51262023-01-16late20.99
    employees60 filas · 8 columnas
    • employee_idinteger
    • first_namevarchar
    • last_namevarchar
    • departmentvarchar
    • titlevarchar
    • manager_idinteger
    • hire_datedate
    • salaryinteger
    Primeras 5 filas de employees
    employee_idfirst_namelast_namedepartmenttitlemanager_idhire_datesalary
    1CatherineHaleExecutiveChief Executive OfficerNaN2016-03-01285000
    2RobertKimExecutiveChief Operating Officer12017-01-09235000
    3MeganCollinsSalesHead of Sales12018-04-16172000
    4ChloeGutierrezSalesAccount Executive32020-01-2172000
    5FrankBellSalesAccount Executive32023-05-2173000
    marketing_spend3,276 filas · 3 columnas
    • spend_datedate
    • channelvarchar
    • amountdecimal
    Primeras 5 filas de marketing_spend
    spend_datechannelamount
    2023-01-01paid492.99
    2023-01-01email37.38
    2023-01-01social200.53
    2023-01-02paid709.13
    2023-01-02email64.13

    La lección

    Por qué revisar primero la división

    Una prueba A/B promete una cosa: los dos grupos solo difieren en lo que cambiaste. Si la asignación está rota, esa promesa desaparece y el resultado no significa nada, por limpio que se vea. Así que antes de mirar la conversión, compruebas que cada grupo recibió la parte del tráfico que debía. Esa revisión tiene nombre: la prueba de desajuste de proporción de muestra, o SRM.

    La prueba del banner de Harbor & Pine dividió las sesiones por session_id: los ids pares vieron el banner (grupo B) y los impares no (grupo A). El plan era 50/50, así que cada grupo debería tener la mitad de las sesiones de la ventana.

    test["arm"].value_counts()
    # A    2604
    # B    2604

    La revisión con chi-cuadrado

    “Cerca de la mitad” no es una prueba. El estadístico chi-cuadrado de bondad de ajuste mide qué tan lejos están los conteos de lo que predice el plan:

    expected = len(test) / 2
    chi2 = (n["A"] - expected) ** 2 / expected + (n["B"] - expected) ** 2 / expected

    Con dos grupos hay un grado de libertad, y un chi-cuadrado con un grado de libertad es el cuadrado de una variable normal estándar. Así puedes obtener su valor p de la distribución normal de la biblioteca estándar de Python, sin SciPy:

    from statistics import NormalDist
    p = 2 * (1 - NormalDist().cdf(math.sqrt(chi2)))

    Aquí los grupos son exactamente iguales, así que chi2 es 0 y el valor p es 1. La división está sana.

    Cómo se ve una revisión que falla

    Supón que el grupo B perdió el 3% de sus sesiones por un script lento del banner: 2,604 en A y 2,526 en B. Entonces chi2 es de unos 1.19 y p de unos 0.28, que todavía pasa. Si escalas la misma pérdida del 3% a una prueba de 50,000 sesiones por grupo, p baja de 0.0001. Las revisiones de SRM se vuelven más estrictas cuanto más grandes son las pruebas, y por eso las empresas grandes las ejecutan de forma automática. La mayoría de los equipos considera un valor p menor que 0.001 como una división fallida y detiene el análisis.

    Errores comunes

    • Mirar los conteos a ojo. 50.6% frente a 49.4% parece bien y puede ser un error real en una prueba grande.
    • Cortar mal la ventana. started_at < "2025-12-31" descarta todo el último día, porque las marcas de tiempo del 31 de diciembre son posteriores a la medianoche. Usa < "2026-01-01" para una fecha de fin inclusiva.
    • Revisar solo después de que el resultado decepciona. Haz la revisión de SRM primero, siempre, diga lo que diga el resultado. Una revisión que solo haces con malas noticias es una forma de elegir la respuesta que te gusta.

    Nota de entrevista

    “¿Qué revisarías antes de confiar en el resultado de una prueba A/B?” La proporción de muestra es la primera respuesta que esperan los entrevistadores, seguida de la ventana de la prueba, la unidad de aleatorización y la definición de la métrica.

    Elige qué puede ejecutarse en este sitio. Puedes cambiarlo cuando quieras desde "Configurar cookies", al pie de cada página.