import matplotlib.pyplot as plt
import numpy as np

# data from https://allisonhorst.github.io/palmerpenguins/

species = (
    "Adelie\n $\\mu=$3700.66g",
    "Chinstrap\n $\\mu=$3733.09g",
    "Gentoo\n $\\mu=5076.02g$",
)
weight_counts = {
    "Below": np.array([70, 31, 58]),
    "Above": np.array([82, 37, 66]),
}
width = 0.5

fig, ax = plt.subplots()
bottom = np.zeros(3)

for boolean, weight_count in weight_counts.items():
    p = ax.bar(species, weight_count, width, label=boolean, bottom=bottom)
    bottom += weight_count

ax.set_title("Number of penguins with above average body mass")
ax.legend(loc="upper right")

plt.show()

import pandas as pd

# data from https://allisonhorst.github.io/palmerpenguins/
df = pd.DataFrame(data={
    "species": [
            "Adelie\n $\\mu=$3700.66g",
            "Chinstrap\n $\\mu=$3733.09g",
            "Gentoo\n $\\mu=5076.02g$",
        ],
        "below": [70, 31, 58],
        "above": [82, 37, 66],
    })
df

df.plot.bar(
    x="species",
    y=[
        "below",
        "above",
    ],
    label=[
        "Below",
        "Above",
    ],
    color=[
        "red",
        "green",
    ],
    rot=0,
    xlabel="Species",
    ylabel="Count",
    title="Number of penguins with above average body mass",
    figsize=(16, 10),
    stacked=True,
)