Aprenda Challenge: Retention Cohort Analyzer

To deepen your understanding of cohort analysis, you will write a Python script that simulates a simple retention cohort analyzer. Cohort analysis is a powerful tool for growth hackers, as it helps you track user retention over time by grouping users based on shared characteristics—in this case, their signup month. You will use a hardcoded pandas DataFrame to represent user data, including user IDs, their signup months, and retention status (where 1 means retained and 0 means not retained). Your task is to calculate the retention rate for each signup month and display the results, giving you practical experience with both pandas and user retention analytics.

Begin by importing the pandas library and creating a DataFrame with the necessary columns. The DataFrame will contain three columns: user_id, signup_month, and retained. Each row represents a user, their signup month, and whether they were retained.


              123456789101112131415161718192021
            
import pandas as pd

# Hardcoded user data for cohort analysis
data = {
    "user_id": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
    "signup_month": ["2024-01", "2024-01", "2024-01", "2024-02", "2024-02", 
                     "2024-02", "2024-03", "2024-03", "2024-03", "2024-03"],
    "retained": [1, 0, 1, 1, 0, 1, 0, 1, 1, 0]
}

df = pd.DataFrame(data)

# Calculate retention rate for each signup month
cohort = df.groupby("signup_month")["retained"].mean().reset_index()
cohort["retention_rate"] = (cohort["retained"] * 100).round(2)

# Print the results
for _, row in cohort.iterrows():
    print(
        f"Signup Month: {row['signup_month']} - Retention Rate: {row['retention_rate']}%"
    )

This script groups users by their signup_month and calculates the average of the retained column for each group, which represents the retention rate. The retention rate is then multiplied by 100 and rounded to two decimal places for readability. Finally, the script prints the retention rate for each signup month, allowing you to see which cohorts are performing better in terms of user retention. By analyzing these results, you can identify trends and make data-driven decisions to improve user engagement and retention strategies.

Tarefa

Swipe to start coding

Write a Python script that:

Imports the pandas library.
Creates a DataFrame with the following columns: user_id, signup_month, and retained.
The DataFrame should have at least 8 rows, with at least two unique values for signup_month.
Calculates the retention rate (mean of retained) for each signup month.
Prints each signup month and its retention rate as a percentage, rounded to two decimal places.

Solução

Tudo estava claro?

Obrigado pelo seu feedback!

Seção 2. Capítulo 7

single

Pergunte à IA

Pergunte o que quiser ou experimente uma das perguntas sugeridas para iniciar nosso bate-papo

Deslize para mostrar o menu


              123456789101112131415161718192021
            
import pandas as pd

# Hardcoded user data for cohort analysis
data = {
    "user_id": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
    "signup_month": ["2024-01", "2024-01", "2024-01", "2024-02", "2024-02", 
                     "2024-02", "2024-03", "2024-03", "2024-03", "2024-03"],
    "retained": [1, 0, 1, 1, 0, 1, 0, 1, 1, 0]
}

df = pd.DataFrame(data)

# Calculate retention rate for each signup month
cohort = df.groupby("signup_month")["retained"].mean().reset_index()
cohort["retention_rate"] = (cohort["retained"] * 100).round(2)

# Print the results
for _, row in cohort.iterrows():
    print(
        f"Signup Month: {row['signup_month']} - Retention Rate: {row['retention_rate']}%"
    )