快捷導(dǎo)航

Python利用pandas和matplotlib實(shí)現(xiàn)繪制堆疊柱狀圖

更新時(shí)間：2023年11月14日 08:48:37 作者：懸崖上的金魚(yú)

在數(shù)據(jù)可視化中,堆疊柱狀圖是一種常用的圖表類(lèi)型,它能夠清晰地展示多個(gè)類(lèi)別的數(shù)據(jù),本文將演示如何使用 Python 的 pandas 和 matplotlib 庫(kù)繪制優(yōu)化的堆疊柱狀圖,需要的可以參考下

在數(shù)據(jù)可視化中，堆疊柱狀圖是一種常用的圖表類(lèi)型，它能夠清晰地展示多個(gè)類(lèi)別的數(shù)據(jù)，并突出顯示每個(gè)類(lèi)別中各部分的總量和組成比例。本文將演示如何使用 Python 的 pandas 和 matplotlib 庫(kù)繪制優(yōu)化的堆疊柱狀圖，并展示了銷(xiāo)售數(shù)量隨店鋪名稱(chēng)變化的情況。

導(dǎo)入必要的庫(kù)

首先，我們需要導(dǎo)入 pandas 和 matplotlib.pyplot 庫(kù)，并指定中文字體為黑體，代碼如下：

import pandas as pd
import matplotlib.pyplot as plt

plt.rcParams['font.family'] = ['SimHei']  # 指定中文字體為黑體

讀取數(shù)據(jù)

接下來(lái)，我們使用 pandas 的 read_excel 函數(shù)讀取 Excel 文件中的數(shù)據(jù)，并指定讀取的工作表名稱(chēng)為“Sheet3”，如下所示：

df = pd.read_excel(r'C:\Users\liuchunlin2\Desktop\新建文件夾\新建 XLSX 工作表.xlsx', sheet_name='Sheet3')

設(shè)置圖形參數(shù)

在繪制堆疊柱狀圖之前，我們需要設(shè)置柱狀圖的寬度和 x 軸的位置，代碼如下：

bar_width = 0.35  # 設(shè)置柱狀圖的寬度
x = df.index  # 設(shè)置x軸的位置

繪制堆疊柱狀圖

使用 matplotlib 庫(kù)的 subplots 函數(shù)創(chuàng)建圖形對(duì)象，并使用 bar 函數(shù)繪制堆疊柱狀圖，具體代碼如下：

fig, ax = plt.subplots()
rects1 = ax.bar(x, df['銷(xiāo)售數(shù)量'], bar_width, label='銷(xiāo)售數(shù)量')
rects2 = ax.bar(x, df['銷(xiāo)售數(shù)量2'], bar_width, bottom=df['銷(xiāo)售數(shù)量'], label='銷(xiāo)售數(shù)量2')

添加標(biāo)簽和標(biāo)題

我們?yōu)閳D形添加軸標(biāo)簽、標(biāo)題、刻度和圖例，使其更具可讀性，具體代碼如下：

ax.set_xlabel('店鋪名稱(chēng)')
ax.set_ylabel('銷(xiāo)售數(shù)量')
ax.set_title('Stacked Bar Chart')
ax.set_xticks(x)
ax.set_xticklabels(df['店鋪名稱(chēng)'])
ax.legend()

顯示數(shù)據(jù)標(biāo)簽

最后，我們使用 annotate 函數(shù)在每個(gè)柱子上方顯示數(shù)據(jù)標(biāo)簽，以展示具體的銷(xiāo)售數(shù)量，具體代碼如下：

for rect in rects1:
    height = rect.get_height()
    ax.annotate(f'{height}', xy=(rect.get_x() + rect.get_width() / 2, height), xytext=(0, 3),
                textcoords='offset points', ha='center', va='bottom')

for rect1, rect2 in zip(rects1, rects2):
    height1 = rect1.get_height()
    height2 = rect2.get_height()
    total_height = height1 + height2
    ax.annotate(f'{height2}', xy=(rect2.get_x() + rect2.get_width() / 2, total_height), xytext=(0, 3),
                textcoords='offset points', ha='center', va='bottom')

顯示圖形

最后，使用 plt.show() 函數(shù)顯示繪制好的堆疊柱狀圖，代碼如下：

plt.show()

通過(guò)以上步驟，我們成功繪制出了堆疊柱狀圖，展示了不同店鋪的銷(xiāo)售數(shù)量情況。

圖表效果圖展示

完整代碼

import pandas as pd
import matplotlib.pyplot as plt

plt.rcParams['font.family'] = ['SimHei']  # 指定中文字體為黑體
# 讀取Excel文件
df = pd.read_excel(r'C:\Users\liuchunlin2\Desktop\新建文件夾\新建 XLSX 工作表.xlsx', sheet_name='Sheet3')
# 設(shè)置柱狀圖的寬度
bar_width = 0.35
# 設(shè)置x軸的位置
x = df.index

# 繪制堆疊柱狀圖
fig, ax = plt.subplots()
rects1 = ax.bar(x, df['銷(xiāo)售數(shù)量'], bar_width, label='銷(xiāo)售數(shù)量')
rects2 = ax.bar(x, df['銷(xiāo)售數(shù)量2'], bar_width, bottom=df['銷(xiāo)售數(shù)量'], label='銷(xiāo)售數(shù)量2')

# 添加標(biāo)簽和標(biāo)題
ax.set_xlabel('店鋪名稱(chēng)')
ax.set_ylabel('銷(xiāo)售數(shù)量')
ax.set_title('Stacked Bar Chart')
ax.set_xticks(x)
ax.set_xticklabels(df['店鋪名稱(chēng)'])
ax.legend()

# 在每個(gè)柱子上方顯示數(shù)據(jù)標(biāo)簽
for rect in rects1:
    height = rect.get_height()
    ax.annotate(f'{height}', xy=(rect.get_x() + rect.get_width() / 2, height), xytext=(0, 3),
                textcoords='offset points', ha='center', va='bottom')

for rect1, rect2 in zip(rects1, rects2):
    height1 = rect1.get_height()
    height2 = rect2.get_height()
    total_height = height1 + height2
    ax.annotate(f'{height2}', xy=(rect2.get_x() + rect2.get_width() / 2, total_height), xytext=(0, 3),
                textcoords='offset points', ha='center', va='bottom')

# 顯示圖形
plt.show()

到此這篇關(guān)于Python利用pandas和matplotlib實(shí)現(xiàn)繪制堆疊柱狀圖的文章就介紹到這了,更多相關(guān)Python堆疊柱狀圖內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: