Goodness of fit evaluation in SciPy - Mini Project: Build & Apply

Goodness of Fit Evaluation with SciPy

📖 Scenario: You are a data analyst working with a small dataset of observed counts from a survey. You want to check if these observed counts fit a theoretical expected distribution using a goodness of fit test.

🎯 Goal: Build a Python program that uses SciPy to perform a chi-square goodness of fit test comparing observed data to expected data.

📋 What You'll Learn

Create a list called observed_counts with exact values [18, 22, 20, 15, 25]

Create a list called expected_counts with exact values [20, 20, 20, 20, 20]

Use scipy.stats.chisquare to perform the goodness of fit test

Print the chi-square statistic and p-value with clear labels

💡 Why This Matters

🌍 Real World

Goodness of fit tests are used in surveys, quality control, and scientific experiments to check if data matches expected patterns.

💼 Career

Data analysts and scientists use these tests to validate models and assumptions about data distributions.

1

Create observed data list

Create a list called observed_counts with these exact values: [18, 22, 20, 15, 25].

SciPy

# Create the observed counts list
# Your code here

Hint

Use square brackets to create a list and separate numbers with commas.

2

Create expected data list

Create a list called expected_counts with these exact values: [20, 20, 20, 20, 20].

SciPy

observed_counts = [18, 22, 20, 15, 25]
# Create the expected counts list
# Your code here

Hint

Use the same list syntax as before to create the expected counts list.

3

Perform chi-square goodness of fit test

Import chisquare from scipy.stats and use it to perform a chi-square test comparing observed_counts and expected_counts. Store the result in a variable called chi2_result.

SciPy

observed_counts = [18, 22, 20, 15, 25]
expected_counts = [20, 20, 20, 20, 20]
# Import chisquare and perform the test
# Your code here

Hint

Use chisquare(f_obs=observed_counts, f_exp=expected_counts) to run the test.

4

Print chi-square statistic and p-value

Print the chi-square statistic and p-value from chi2_result with labels: Chi-square statistic: and p-value:.

SciPy

from scipy.stats import chisquare

observed_counts = [18, 22, 20, 15, 25]
expected_counts = [20, 20, 20, 20, 20]

chi2_result = chisquare(f_obs=observed_counts, f_exp=expected_counts)
# Print the chi-square statistic and p-value
# Your code here

Hint

Use print(f"Chi-square statistic: {chi2_result.statistic}") and similarly for p-value.

Practice

1. What does the chi-square goodness of fit test in scipy.stats.chisquare primarily evaluate?

easy

A. The mean difference between two samples

B. How well observed data matches expected frequencies

C. The correlation between two variables

D. The variance within a single dataset

Solution

1. Step 1: Understand the purpose of chi-square test

   The chi-square goodness of fit test compares observed data frequencies to expected frequencies to check if they match.

2. Step 2: Identify what scipy.stats.chisquare does

   This function calculates the chi-square statistic and p-value to evaluate the fit between observed and expected counts.

3. Final Answer:

   How well observed data matches expected frequencies -> Option B

4. Quick Check:

   Goodness of fit = observed vs expected match [OK]

Hint: Chi-square tests observed vs expected frequencies [OK]

Common Mistakes:

• Confusing goodness of fit with correlation
• Thinking it measures mean differences
• Mixing variance analysis with goodness of fit

2. Which of the following is the correct way to import the chi-square goodness of fit test function from scipy?

easy

A. import scipy.chisquare

B. from scipy import chisquare

C. from scipy.stats import chisquare

D. import scipy.stats.chisquare as cs

Solution

1. Step 1: Recall the module structure of scipy

   The chi-square test function is inside the stats submodule of scipy.

2. Step 2: Identify correct import syntax

   The correct import is from scipy.stats import chisquare to directly access the function.

3. Final Answer:

   from scipy.stats import chisquare -> Option C

4. Quick Check:

   Correct import = from scipy.stats import chisquare [OK]

Hint: Import from scipy.stats for statistical tests [OK]

Common Mistakes:

• Trying to import chisquare directly from scipy
• Using incorrect module paths
• Using alias without import

3. What will be the output of the following code?

from scipy.stats import chisquare
observed = [20, 30, 50]
expected = [25, 25, 50]
result = chisquare(f_obs=observed, f_exp=expected)
print(round(result.statistic, 2), round(result.pvalue, 3))

medium

A. 2.0 0.368

B. 3.0 0.223

C. 0.5 0.778

D. 1.0 0.607

Solution

1. Step 1: Calculate chi-square statistic manually

   Chi-square = sum((observed - expected)^2 / expected) = ((20-25)^2/25) + ((30-25)^2/25) + ((50-50)^2/50) = (25/25)+(25/25)+0 = 1+1+0 = 2.0

2. Step 2: Interpret p-value from scipy output

   Using scipy.stats.chisquare with these values gives a p-value around 0.368, indicating moderate fit.

3. Final Answer:

   2.0 0.368 -> Option A

4. Quick Check:

   Chi-square stat = 2.0, p-value ≈ 0.368 [OK]

Hint: Calculate chi-square stat then check p-value [OK]

Common Mistakes:

• Forgetting to square differences
• Dividing by wrong expected values
• Mixing up statistic and p-value

4. Identify the error in this code snippet for performing a chi-square goodness of fit test:

from scipy.stats import chisquare
observed = [15, 25, 35]
expected = [20, 20]
result = chisquare(f_obs=observed, f_exp=expected)
print(result)

medium

A. Observed and expected arrays have different lengths

B. chisquare function is not imported correctly

C. Expected frequencies must be integers

D. Missing p-value extraction from result

Solution

1. Step 1: Check input array lengths

   The observed array has 3 elements, but expected has only 2 elements, which is invalid for chi-square test.

2. Step 2: Understand scipy requirement

   Both observed and expected arrays must be the same length to compare frequencies correctly.

3. Final Answer:

   Observed and expected arrays have different lengths -> Option A

4. Quick Check:

   Array length mismatch causes error [OK]

Hint: Observed and expected must be same length [OK]

Common Mistakes:

• Ignoring length mismatch
• Assuming expected must be integers
• Thinking import or print is the error

5. You have observed counts of [40, 35, 25] for three categories. You expect them to be equally likely. Using scipy.stats.chisquare, what is the p-value indicating if the observed data fits the equal distribution? (Hint: expected counts are equal for all categories.)

hard

A. 0.789

B. 0.223

C. 0.456

D. 0.174

Solution

1. Step 1: Calculate expected counts for equal distribution

   Total counts = 40+35+25 = 100. Expected counts = [100/3, 100/3, 100/3] ≈ [33.33, 33.33, 33.33].

2. Step 2: Perform chi-square test with scipy

   Using chisquare(f_obs=[40,35,25], f_exp=[33.33,33.33,33.33]) gives a chi-square statistic ≈ 3.5 and p-value ≈ 0.174.

3. Final Answer:

   0.174 -> Option D

4. Quick Check:

   Unequal counts vs equal expected gives p-value ≈ 0.174 [OK]

Hint: Equal expected counts = total/number categories [OK]

Common Mistakes:

• Using observed counts as expected
• Not dividing total counts equally
• Misinterpreting p-value significance