如何在 matplotlib 中合并子图？

Question

I am getting a mongodb query (not sure whether that bit is relevant) and trying to plot a stacked histogram based the values of one of the columns.

cursor = db.gadgets.find()
df = pd.DataFrame(cursor)
df['created'].hist(by=df['gadgetTypeId'], sharex=True, sharey=True, figsize=(16,10), legend=True)

So far so good, but it creates a bunch of subplots. I want one stacked subplot.

Tried to get this using matplotlib.pyplot.hist() , but couldn't find how to use the by argument.

Answer 1

With the following toy dataframe:

import pandas as pd

df = pd.DataFrame(
    {
        "gadgetTypeId": [1, 2, 3, 3, 1, 2, 2, 1, 2, 1],
        "created": [2017, 2018, 2018, 2019, 2017, 2018, 2017, 2017, 2019, 2019],
    }
)

Here is one way to do it:

import matplotlib.pyplot as plt

plt.hist(
    x=[df.loc[df["gadgetTypeId"] == i, "created"] for i in df["gadgetTypeId"].unique()],
    stacked=True,
    color=["r", "g", "b"],
    label=df["gadgetTypeId"].unique(),
)
plt.xticks(ticks=df["created"])
plt.yticks(ticks=range(df["created"].value_counts().max() + 1))
plt.legend()

Which outputs: