From 7bb8be46150e879d0308c210eed58cdee30d5670 Mon Sep 17 00:00:00 2001 From: Ruth Nainggolan Date: Tue, 21 Dec 2021 18:34:59 -0500 Subject: [PATCH] Update categorical_variables.py Add explanatory comments to the example on plotting categorical variables --- examples/lines_bars_and_markers/categorical_variables.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/examples/lines_bars_and_markers/categorical_variables.py b/examples/lines_bars_and_markers/categorical_variables.py index 01863b3c193c..2fd336330549 100644 --- a/examples/lines_bars_and_markers/categorical_variables.py +++ b/examples/lines_bars_and_markers/categorical_variables.py @@ -8,10 +8,12 @@ """ import matplotlib.pyplot as plt +# Example data data = {'apple': 10, 'orange': 15, 'lemon': 5, 'lime': 20} names = list(data.keys()) values = list(data.values()) +# Plot example on three separate subplots with a shared y-axis fig, axs = plt.subplots(1, 3, figsize=(9, 3), sharey=True) axs[0].bar(names, values) axs[1].scatter(names, values) @@ -22,10 +24,12 @@ ############################################################################### # This works on both axes: +# Example data cat = ["bored", "happy", "bored", "bored", "happy", "bored"] dog = ["happy", "happy", "happy", "happy", "bored", "bored"] activity = ["combing", "drinking", "feeding", "napping", "playing", "washing"] +# Plot example with just one subplot fig, ax = plt.subplots() ax.plot(activity, dog, label="dog") ax.plot(activity, cat, label="cat")