Matplotlibのグラフスタイルを変える

 Pythonのグラフ描画ライブラリMatplotlibにはグラフのスタイルがいくつか用意されていて、グラフの配色を変えてみたいときなどに利用できる。スタイルの変更方法を調べたのでまとめておく。

環境

Dockerコンテナ上のJupyter Lab。
# python -V Python 3.7.6 # pip list | grep -e numpy -e matplotlib -e jupyter jupyter 1.0.0 jupyter-client 5.3.4 jupyter-console 6.1.0 jupyter-core 4.6.1 jupyterlab 1.2.6 jupyterlab-server 1.0.6 matplotlib 3.1.3 numpy 1.18.1

スタイルの一覧

使用できるスタイルの一覧はmatplotlib.pyplot.style.availableで確認できる。

# python >>> import matplotlib.pyplot as plt >>> plt.style.available ['fivethirtyeight', 'seaborn-pastel', 'classic', 'Solarize_Light2', 'bmh', 'seaborn-colorblind', 'seaborn-muted', 'seaborn-notebook', 'grayscale', 'seaborn-dark-palette', '_classic_test', 'dark_background', 'fast', 'seaborn-deep', 'seaborn-white', 'seaborn-talk', 'seaborn', 'seaborn-bright', 'seaborn-poster', 'seaborn-darkgrid', 'seaborn-whitegrid', 'seaborn-paper', 'ggplot', 'tableau-colorblind10', 'seaborn-ticks', 'seaborn-dark']

Style sheets referenceには各スタイルのサンプルグラフがある。

スタイルの変更

スタイルを変更してみる。まずはデフォルトのスタイルでグラフを表示。

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

plt.style.use('default')
plt.plot(np.sin(np.linspace(0, 2 * np.pi)), 'o')
plt.show()

つづいて、スタイルを変更してみる。

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

plt.style.use('ggplot')
plt.plot(np.sin(np.linspace(0, 2 * np.pi)), 'o')
plt.show()

いくつかグラフを作成するうちのひとつだけ変えたい場合は、with構文を使うと一時的にスタイルを変えることができる。

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

with plt.style.context('ggplot'):  
    plt.plot(np.sin(np.linspace(0, 2 * np.pi)), 'o')
    plt.show()

すべてのスタイルのグラフを表示

すべてのスタイルのグラフを表示してみる。

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

for style in plt.style.available:
    print('\n[{}]'.format(style))
    with plt.style.context(style):
        plt.plot(np.sin(np.linspace(0, 2 * np.pi)), 'o')
        plt.show()

<以下省略>