Matplotlibによるグラフ作成 第3回

Matplotlibによるグラフ作成 第3回

PythonのMatplotlibを使用した基本的なグラフ作成方法を紹介する
今回は、変化を比較する折れ線グラフの作成方法を紹介する

サンプルデータ生成

指数関数の波形を用意する

import matplotlib.pyplot as plt
import numpy as np

data_s = []
for i in range(20):
    data_s += [[i * np.exp(-j) for j in np.arange(0, 1, 0.05)]]

重ね書き

plot を使用すると、折れ線グラフになる

fig = plt.figure(figsize=(6, 5))
ax = fig.add_subplot(111)

for data in data_s:
    ax.plot(data)

ax.grid()
fig.tight_layout()

マーカー/ライン

fig = plt.figure(figsize=(6, 5))
ax = fig.add_subplot(111)

ax.plot(data_s[1], 'r-',   label='r-')
ax.plot(data_s[6], 'b--',  label='b--')
ax.plot(data_s[11], 'g.-',  label='g.-')
ax.plot(data_s[16], 'y*--', label='y*--')

ax.legend()
ax.grid()
fig.tight_layout()

カラーグラデーション

matplotlib.cm を使用する

import matplotlib.cm as cm

fig = plt.figure(figsize=(6, 5))
ax = fig.add_subplot(111)

N = len(data_s) - 1
for i, data in enumerate(data_s):
    ax.plot(data, color=cm.jet(float(i) / N))

ax.grid()
fig.tight_layout()