patsy_dataframe_to_design_matrices_with_r_style_formulas.py

1import pandas as pd
2import numpy as np
3from patsy import dmatrices
4
5# Create a dummy dataset
6data = pd.DataFrame({
7    "y": [1, 2, 3, 4, 5],
8    "x1": [10, 20, 30, 40, 50],
9    "x2": ["A", "B", "A", "B", "A"]
10})
11
12# Use patsy to create design matrices
13# The formula 'y ~ x1 + x2' means 'y' is the dependent variable, 
14# and 'x1' and 'x2' are the predictors.
15y, X = dmatrices("y ~ x1 + x2", data)
16
17print("Outcome (y):")
18print(y)
19print("\nDesign Matrix (X):")
20print(X)