如何在 matplotlib 中合並子圖？

Question

我得到一個 mongodb 查詢（不確定該位是否相關）並嘗試 plot 基於其中一列的值的堆疊直方圖。

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

到目前為止一切順利，但它創建了一堆子圖。 我想要一個堆疊的子圖。

試圖使用matplotlib.pyplot.hist()得到這個，但找不到如何使用by參數。

Answer 1

用以下玩具 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],
    }
)

這是一種方法：

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()

哪個輸出：