首頁 > 後端開發 > Python教學 > 如何使用 Pandas 和 Matplotlib 在 Python 中建立聚類堆疊長條圖?

如何使用 Pandas 和 Matplotlib 在 Python 中建立聚類堆疊長條圖?

Linda Hamilton
發布: 2024-11-02 12:46:30
原創
358 人瀏覽過

How to Create Clustered Stacked Bar Plots in Python with Pandas and Matplotlib?

建立聚類堆疊條形圖

問題:

您有多個具有相同欄位和索引的資料框,並且想要建立一個堆疊條形圖,其中每個資料框的資料聚集在一起。

使用Pandas 和Matplotlib 的解決方案:

  1. 將資料框合併為單一資料框.
  2. 使用資料框合併為單一資料框.

    • 使用資料框合併為單一資料框.
    • dfall:要繪製的資料框清單。
    • labels:資料框名稱清單(可選)。
  3. title:繪圖的標題(可選)。

H:用於每個資料框的填滿圖案(可選)。

<code class="python">import pandas as pd
import matplotlib.cm as cm
import numpy as np
import matplotlib.pyplot as plt

def plot_clustered_stacked(dfall, labels=None, title=&quot;multiple stacked bar plot&quot;,  H=&quot;/&quot;, **kwargs):
    n_df = len(dfall)
    n_col = len(dfall[0].columns)
    n_ind = len(dfall[0].index)
    axe = plt.subplot(111)

    for df in dfall : # for each data frame
        axe = df.plot(kind=&quot;bar&quot;,
                      linewidth=0,
                      stacked=True,
                      ax=axe,
                      legend=False,
                      grid=False,
                      **kwargs)  # make bar plots

    h,l = axe.get_legend_handles_labels() # get the handles we want to modify
    for i in range(0, n_df * n_col, n_col): # len(h) = n_col * n_df
        for j, pa in enumerate(h[i:i+n_col]):
            for rect in pa.patches: # for each index
                rect.set_x(rect.get_x() + 1 / float(n_df + 1) * i / float(n_col))
                rect.set_hatch(H * int(i / n_col)) #edited part
                rect.set_width(1 / float(n_df + 1))

    axe.set_xticks((np.arange(0, 2 * n_ind, 2) + 1 / float(n_df + 1)) / 2.)
    axe.set_xticklabels(df.index, rotation = 0)
    axe.set_title(title)

    # Add invisible data to add another legend
    n=[]
    for i in range(n_df):
        n.append(axe.bar(0, 0, color=&quot;gray&quot;, hatch=H * i))

    l1 = axe.legend(h[:n_col], l[:n_col], loc=[1.01, 0.5])
    if labels is not None:
        l2 = plt.legend(n, labels, loc=[1.01, 0.1])
    axe.add_artist(l1)
    return axe

# create fake dataframes
df1 = pd.DataFrame(np.random.rand(4, 5),
                   index=[&quot;A&quot;, &quot;B&quot;, &quot;C&quot;, &quot;D&quot;],
                   columns=[&quot;I&quot;, &quot;J&quot;, &quot;K&quot;, &quot;L&quot;, &quot;M&quot;])
df2 = pd.DataFrame(np.random.rand(4, 5),</code>
登入後複製
程式碼:

以上是如何使用 Pandas 和 Matplotlib 在 Python 中建立聚類堆疊長條圖?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板