Welcome to our exciting lesson! We shall embark on learning and mastering Hypothesis Testing using Python. It might sound complicated, but it’s like deciding if a toy is worth buying based on its reviews. We'll focus on the T-test, a way to tell if two groups are different.
Python has useful tools, Scipy
and Statsmodels
, which help us do these tests quickly and accurately. By the end, you'll understand Hypothesis Testing, know what a T-test is, and be able to do a T-test using Python. So, let's start!
A hypothesis is a guess about a group. For example, "adult males in the U.S. average 180 lbs." In Hypothesis Testing, we try to prove or disprove these guesses using collected data.
Null Hypothesis (H0): The guess we're challenging. For example, "adult males in the U.S. do not average 180 lbs."
Alternative Hypothesis (HA): The guess we're trying to prove (e.g., "Adult males in the U.S. average 180 lbs.").
Think of it like a courtroom trial. The null hypothesis is on trial, and the alternative hypothesis offers the evidence.
Let's understand the T-test better. It checks if the two groups' mean values are truly different. It's like testing if two pots of coffee are different temperatures due to one being under an AC vent or just by chance.
There are three main types of T-tests:
T-test gives us two values: the t-statistic
and the p-value
. The t-statistic
represents the size of the difference relative to the variation in your sample data. Put simply, the bigger the t-statistic
, the more difference there is between groups mean values. The p-value
is the probability that the results could be random (i.e., happened by chance). If the p-value
is less than 0.05, usually, we conclude that the difference is statistically significant and not due to randomness.
Python has powerful tools, Scipy
and Statsmodels
, for Hypothesis Testing.
For example, to do a one-sample T-test in Python, we can use Scipy
's ttest_1samp()
function.
Let's begin by assuming that the null hypothesis is that the mean age of users (provided as the ages
array) equals 30
. Therefore, the alternative hypothesis states that the mean age is not 30
. Let’s illustrate how we can test this:
Python1import numpy as np 2from scipy import stats 3 4ages = np.array([25, 33, 26, 25, 27, 27, 27, 29, 30, 31, 33]) # mean = 28.45 5t_statistic, p_value = stats.ttest_1samp(ages, 30) 6print("t-statistic:", t_statistic) # -1.74 7print("p-value:", p_value) # 0.1124
In this case, we fail to reject the null hypothesis because the p-value is greater than 0.05
(the conventional cutoff). It means that we don't have enough statistical evidence to claim that the mean age of users is different from 30
.
Now let's modify our numpy array to contain a normally distributed sample with a mean that differs from 30
:
Python1import numpy as np 2from scipy import stats 3 4ages = np.random.normal(loc=33, scale=5, size=90) # mean = 33 5t_statistic, p_value = stats.ttest_1samp(ages, 30) 6print("t-statistic:", t_statistic) # 4.872 7print("p-value:", p_value) # ~0.000
We might reject the null hypothesis in this case as the p_value
is less than 0.05
. It suggests strong evidence against the null hypothesis, implying that the average age of users is significantly different from 30
.
Imagine you want to test if two teams in your office work the same hours. After collecting data, you can use a two-sample T-test to find out.
We will use the stats.ttest_ind
function for the two-sample T-test. Here’s an example:
Python1import numpy as np 2from scipy import stats 3 4team_A_hours = np.array([8.5, 7.5, 8, 8, 8, 8, 8, 8.5, 9]) 5team_B_hours = np.array([9, 8, 9, 9, 9, 9, 9, 9, 9.5]) 6t_statistic, p_value = stats.ttest_ind(team_A_hours, team_B_hours) 7print("t-statistic:", t_statistic) # -4 8print("p-value:", p_value) # 0.00103
The p-value
is less than 0.05
, so we can reject the null hypothesis, meaning we have sufficient statistical evidence to say that there's a significant difference between the mean working hours of Teams A and B.
Well done! We've learned Hypothesis Testing, understood T-tests, and done a T-test in Python. T-tests are a helpful way to make decisions with data.
Now it's time for you to practice. The more you practice, the better you'll understand. Let's dive into some data with Python!