Matplotlib による積み上げ棒グラフ

はじめに

この実験では、Python の Matplotlib ライブラリを使って積み上げ棒グラフを作成する方法を学びます。平均体重を超えるペンギンの数を示す積み上げ棒グラフを作成するために、ペンギンデータを使用します。

VM のヒント

VM の起動が完了したら、左上隅をクリックしてノートブックタブに切り替え、Jupyter Notebook を使って練習しましょう。

時々、Jupyter Notebook が読み込み終了するまで数秒待つ必要がある場合があります。Jupyter Notebook の制限により、操作の検証を自動化することはできません。

学習中に問題に遭遇した場合は、Labby にお問い合わせください。セッション後にフィードバックを提供してください。すぐに問題を解決いたします。

ライブラリのインポート

必要なライブラリをインポートして始めましょう。その中には、numpy と matplotlib.pyplot が含まれます。

import matplotlib.pyplot as plt
import numpy as np

データの定義

積み上げ棒グラフを作成するために使用するデータを定義します。

species = (
    "Adelie\n $\\mu=$3700.66g",
    "Chinstrap\n $\\mu=$3733.09g",
    "Gentoo\n $\\mu=5076.02g$",
)
weight_counts = {
    "Below": np.array([70, 31, 58]),
    "Above": np.array([82, 37, 66]),
}
width = 0.5

積み上げ棒グラフを作成する

matplotlib.pyplot.bar を使って積み上げ棒グラフを作成し、各体重カテゴリをループして棒を積み上げます。

fig, ax = plt.subplots()
bottom = np.zeros(3)

for boolean, weight_count in weight_counts.items():
    p = ax.bar(species, weight_count, width, label=boolean, bottom=bottom)
    bottom += weight_count

ax.set_title("Number of penguins with above average body mass")
ax.legend(loc="upper right")

グラフを表示する

matplotlib.pyplot.show() を使って積み上げ棒グラフを表示します。

plt.show()

まとめ

この実験では、Python の Matplotlib ライブラリを使って積み上げ棒グラフを作成する方法を学びました。まず必要なライブラリをインポートし、グラフを作成するために使用するデータを定義し、そして matplotlib.pyplot.bar を使って積み上げ棒グラフを作成しました。最後に、matplotlib.pyplot.show() を使ってグラフを表示しました。