matplotlib: 折れ線グラフのリアルタイムプロットでラベルやカラーなどの属性を変える（音声信号でのデモ付き）

f:id:Kurene:20210601200449g:plain

前回の記事では、matplotlibでの折れ線グラフのプロット画像のラベルや属性を変更しました。

今回は、折れ線グラフのリアルタイムプロットでラベルやカラーなどの属性を変えるコードを実装をしました。

import numpy as np
import matplotlib.pyplot as plt


length = 100
x = np.arange(0, length)
y = np.array([
    np.random.normal(0, 5.0, length),
    np.random.normal(0, 1.0, length),
    np.random.normal(0, 3.0, length),
])
colors = ['r', 'g', 'b']
labels = ["", "", ""]
markers = ["o", "x", "*"]
alphas = [0.25, 0.5, 0.75]

plt.ion()
fig = plt.figure(figsize=(5,5))
ax = fig.add_subplot(111)
ax.set_prop_cycle(color=colors, marker=markers, alpha=alphas)

lines = ax.plot(x, y.T, "-*") 
plt.legend(lines, labels, loc='upper right')
plt.ylim(-10, 10)
plt.grid()
plt.show()


from collections import deque
colors = deque(colors)

for n in range(0, 50):
    y = np.array([
        np.random.normal(0, 5.0, length),
        np.random.normal(0, 1.0, length),
        np.random.normal(0, 3.0, length),
    ])
    labels = [f'plot{n}_no_1', f'plot{n}_no_2', f'plot{n}_no_3']
    colors.rotate(1)
    
    
    for k in range(0,len(lines)):
        lines[k].set_ydata(y[k,:]) 
        lines[k].set_color(colors[k])
        lines[k].set_marker(markers[k])

    plt.legend(lines, labels, loc='upper right')
    plt.pause(0.1)