1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
#!/usr/bin/env python
# coding: utf-8
# DO NOT EDIT
# Autogenerated from the notebook categorical_interaction_plot.ipynb.
# Edit the notebook and then sync the output with this file.
#
# flake8: noqa
# DO NOT EDIT
# # Plot Interaction of Categorical Factors
# In this example, we will visualize the interaction between categorical
# factors. First, we will create some categorical data. Then, we will plot
# it using the interaction_plot function, which internally re-codes the
# x-factor categories to integers.
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from statsmodels.graphics.factorplots import interaction_plot
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))
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,
)
|
