[PYTHON] How to draw a bar graph that summarizes multiple series with matplotlib

# version
matplotlib == 3.2.2

import numpy as np
import matplotlib.pyplot as plt

#Vertical bar
def barplot(ax, labels, datas):
    x = np.arange(len(labels))  # the label locations
    width = 0.35  # the width of the bars

    rects1 = ax.bar(x - width/2, datas[0]["val"], width, label=datas[0]["label"])
    rects2 = ax.bar(x + width/2, datas[1]["val"], width, label=datas[1]["label"])

    # Add some text for labels, title and custom x-axis tick labels, etc.
    ax.set_xticks(x)
    ax.set_xticklabels(labels)

#Horizontal bar
def barhplot(ax, labels, datas):
    x = np.arange(len(labels))  # the label locations
    width = 0.35  # the width of the bars

    rects1 = ax.barh(x - width/2, datas[0]["val"], width, label=datas[0]["label"])
    rects2 = ax.barh(x + width/2, datas[1]["val"], width, label=datas[1]["label"])

    # Add some text for labels, title and custom x-axis tick labels, etc.
    ax.set_yticks(x)
    ax.set_yticklabels(labels)
    plt.gca().invert_yaxis()

labels = ['G1', 'G2', 'G3', 'G4', 'G5']
datas = [
    {"label": "men1", "val": [20, 34, 30, 35, 27]},
    {"label": "women2", "val": [25, 32, 34, 20, 25]}]
fig, ax = plt.subplots()
# barhplot(ax, labels, datas)
barplot(ax, labels, datas)
plt.legend()
plt.show()

Recommended Posts