快捷導(dǎo)航

matplotlib bar()實現(xiàn)多組數(shù)據(jù)并列柱狀圖通用簡便創(chuàng)建方法

更新時間：2021年02月24日 11:12:29 作者：mighty13

這篇文章主要介紹了matplotlib bar()實現(xiàn)多組數(shù)據(jù)并列柱狀圖通用簡便創(chuàng)建方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

在使用柱狀圖時，經(jīng)常遇到需要多組數(shù)據(jù)進(jìn)行比較的情況。
繪制單個數(shù)據(jù)系列的柱形圖比較簡單，多組數(shù)據(jù)柱狀圖繪制的關(guān)鍵有三點：

多次調(diào)用bar()函數(shù)即可在同一子圖中繪制多組柱形圖。
為了防止柱子重疊，每個柱子在x軸上的位置需要依次遞增，如果柱子緊挨，這個距離即柱子寬度。
為了使刻度標(biāo)簽居中，需要調(diào)整x軸刻度標(biāo)簽的位置。

由上述可知，多組數(shù)據(jù)并列柱狀圖需要計算柱子x軸上的位置和x軸刻度標(biāo)簽。
因此，有兩種實現(xiàn)方案：

x軸刻度標(biāo)簽位置固定，根據(jù)x軸刻度計算每個柱子的寬度
每個柱子的寬度固定，計算x軸刻度標(biāo)簽位置，使之居中

下面使用第一種方法演示兩組數(shù)據(jù)、三組數(shù)據(jù)、四組數(shù)據(jù)的并列柱狀圖。
使用方法一、方法二演示通用多組并列柱狀圖的創(chuàng)建方法。

兩組數(shù)據(jù)、三組數(shù)據(jù)、四組數(shù)據(jù)的并列柱狀圖

在這里插入圖片描述

import matplotlib
import matplotlib.pyplot as plt
import numpy as np

plt.figure(figsize=(13, 4))
# 構(gòu)造x軸刻度標(biāo)簽、數(shù)據(jù)
labels = ['G1', 'G2', 'G3', 'G4', 'G5']
first = [20, 34, 30, 35, 27]
second = [25, 32, 34, 20, 25]
third = [21, 31, 37, 21, 28]
fourth = [26, 31, 35, 27, 21]

# 兩組數(shù)據(jù)
plt.subplot(131)
x = np.arange(len(labels)) # x軸刻度標(biāo)簽位置
width = 0.25 # 柱子的寬度
# 計算每個柱子在x軸上的位置，保證x軸刻度標(biāo)簽居中
# x - width/2，x + width/2即每組數(shù)據(jù)在x軸上的位置
plt.bar(x - width/2, first, width, label='1')
plt.bar(x + width/2, second, width, label='2')
plt.ylabel('Scores')
plt.title('2 datasets')
# x軸刻度標(biāo)簽位置不進(jìn)行計算
plt.xticks(x, labels=labels)
plt.legend()
# 三組數(shù)據(jù)
plt.subplot(132)
x = np.arange(len(labels)) # x軸刻度標(biāo)簽位置
width = 0.25 # 柱子的寬度
# 計算每個柱子在x軸上的位置，保證x軸刻度標(biāo)簽居中
# x - width，x， x + width即每組數(shù)據(jù)在x軸上的位置
plt.bar(x - width, first, width, label='1')
plt.bar(x, second, width, label='2')
plt.bar(x + width, third, width, label='3')
plt.ylabel('Scores')
plt.title('3 datasets')
# x軸刻度標(biāo)簽位置不進(jìn)行計算
plt.xticks(x, labels=labels)
plt.legend()
# 四組數(shù)據(jù)
plt.subplot(133)
x = np.arange(len(labels)) # x軸刻度標(biāo)簽位置
width = 0.2 # 柱子的寬度
# 計算每個柱子在x軸上的位置，保證x軸刻度標(biāo)簽居中
plt.bar(x - 1.5*width, first, width, label='1')
plt.bar(x - 0.5*width, second, width, label='2')
plt.bar(x + 0.5*width, third, width, label='3')
plt.bar(x + 1.5*width, fourth, width, label='4')
plt.ylabel('Scores')
plt.title('4 datasets')
# x軸刻度標(biāo)簽位置不進(jìn)行計算
plt.xticks(x, labels=labels)
plt.legend()

plt.show()

通用多組并列柱狀圖的簡便創(chuàng)建方法

上面的示例比較簡易，有一些問題沒有考慮。為了便于重復(fù)使用，下面的通用方法可調(diào)整x軸標(biāo)簽刻度步長、每組柱子的總寬度、每組柱子之間的間隙、組與組之間的間隙。

在這里插入圖片描述

方法一

import matplotlib
import matplotlib.pyplot as plt
import numpy as np

label = ['G1', 'G2', 'G3', 'G4', 'G5']
first = [20, 34, 30, 35, 27]
second = [25, 32, 34, 20, 25]
third = [21, 31, 37, 21, 28]
fourth = [26, 31, 35, 27, 21]
data = [first, second, third, fourth]


def create_multi_bars(labels, datas, tick_step=1, group_gap=0.2, bar_gap=0):
  '''
  labels : x軸坐標(biāo)標(biāo)簽序列
  datas ：數(shù)據(jù)集，二維列表，要求列表每個元素的長度必須與labels的長度一致
  tick_step ：默認(rèn)x軸刻度步長為1，通過tick_step可調(diào)整x軸刻度步長。
  group_gap : 柱子組與組之間的間隙，最好為正值，否則組與組之間重疊
  bar_gap ：每組柱子之間的空隙，默認(rèn)為0，每組柱子緊挨，正值每組柱子之間有間隙，負(fù)值每組柱子之間重疊
  '''
  # ticks為x軸刻度
  ticks = np.arange(len(labels)) * tick_step
  # group_num為數(shù)據(jù)的組數(shù)，即每組柱子的柱子個數(shù)
  group_num = len(datas)
  # group_width為每組柱子的總寬度，group_gap 為柱子組與組之間的間隙。
  group_width = tick_step - group_gap
  # bar_span為每組柱子之間在x軸上的距離，即柱子寬度和間隙的總和
  bar_span = group_width / group_num
  # bar_width為每個柱子的實際寬度
  bar_width = bar_span - bar_gap
  # baseline_x為每組柱子第一個柱子的基準(zhǔn)x軸位置，隨后的柱子依次遞增bar_span即可
  baseline_x = ticks - (group_width - bar_span) / 2
  for index, y in enumerate(datas):
    plt.bar(baseline_x + index*bar_span, y, bar_width)
  plt.ylabel('Scores')
  plt.title('multi datasets')
  # x軸刻度標(biāo)簽位置與x軸刻度一致
  plt.xticks(ticks, labels)
  plt.show()
  

create_multi_bars(label, data, bar_gap=0.1)

方法二

import matplotlib
import matplotlib.pyplot as plt
import numpy as np

label = ['G1', 'G2', 'G3', 'G4', 'G5']
first = [20, 34, 30, 35, 27]
second = [25, 32, 34, 20, 25]
third = [21, 31, 37, 21, 28]
fourth = [26, 31, 35, 27, 21]
data = [first, second, third, fourth]


def create_multi_bars(labels, datas, tick_step=1, group_gap=0.2, bar_gap=0):
  '''
  labels : x軸坐標(biāo)標(biāo)簽序列
  datas ：數(shù)據(jù)集，二維列表，要求列表每個元素的長度必須與labels的長度一致
  tick_step ：默認(rèn)x軸刻度步長為1，通過tick_step可調(diào)整x軸刻度步長。
  group_gap : 柱子組與組之間的間隙，最好為正值，否則組與組之間重疊
  bar_gap ：每組柱子之間的空隙，默認(rèn)為0，每組柱子緊挨，正值每組柱子之間有間隙，負(fù)值每組柱子之間重疊
  '''
  # x為每組柱子x軸的基準(zhǔn)位置
  x = np.arange(len(labels)) * tick_step
  # group_num為數(shù)據(jù)的組數(shù)，即每組柱子的柱子個數(shù)
  group_num = len(datas)
  # group_width為每組柱子的總寬度，group_gap 為柱子組與組之間的間隙。
  group_width = tick_step - group_gap
  # bar_span為每組柱子之間在x軸上的距離，即柱子寬度和間隙的總和
  bar_span = group_width / group_num
  # bar_width為每個柱子的實際寬度
  bar_width = bar_span - bar_gap
  # 繪制柱子
  for index, y in enumerate(datas):
    plt.bar(x + index*bar_span, y, bar_width)
  plt.ylabel('Scores')
  plt.title('multi datasets')
  # ticks為新x軸刻度標(biāo)簽位置，即每組柱子x軸上的中心位置
  ticks = x + (group_width - bar_span) / 2
  plt.xticks(ticks, labels)
  plt.show()

create_multi_bars(label, data[:3], bar_gap=0.1)

您可能感興趣的文章: