Example: Plot Interaction of Categorical Factors
Plot Interaction of Categorical Factors
In this example, we will vizualize the interaction between categorical factors. First, we will create some categorical data are initialized. Then plotted using the interaction_plot function which internally recodes the x-factor categories to ingegers.
In [1]:
%matplotlib inline import numpy as np import matplotlib.pyplot as plt import pandas as pd from statsmodels.graphics.factorplots import interaction_plot
In [2]:
np.random.seed(12345) weight = pd.Series(np.repeat(['low', 'hi', 'low', 'hi'], 15), name='weight') nutrition = pd.Series(np.repeat(['lo_carb', 'hi_carb'], 30), name='nutrition') days = np.log(np.random.randint(1, 30, size=60))
In [3]:
fig, ax = plt.subplots(figsize=(6, 6)) fig = interaction_plot(x=weight, trace=nutrition, response=days, colors=['red', 'blue'], markers=['D', '^'], ms=10, ax=ax)
© 2009–2012 Statsmodels Developers
© 2006–2008 Scipy Developers
© 2006 Jonathan E. Taylor
Licensed under the 3-clause BSD License.
http://www.statsmodels.org/stable/examples/notebooks/generated/categorical_interaction_plot.html