pythonで横軸に時間、縦軸に電極とした図を作りたい

Question

表題の通りなのですが、ラスター図を作りたいです。

ラスター図というのは、ニューロンからの計測データに対して、
横軸に計測時間、縦軸に電極とした図です。

もう少し状況を詳しく書くと、
いま、時間が0~100秒まであるとして、
1秒ごとに発火した(1)、発火しない(0)のデータが電極数(10)だけあるとします。

電極1 [0,1,1,...,0]
電極2 [0,1,0,...,1]
みたいな感じです。

これを、横軸に0~100(s)、縦軸に電極1,2...,10として、
0のところは何もせず、1のところは黒丸などをつけるとして、
一斉に表示したいです。

どうすればよいでしょうか？
あまり時系列に使うイメージはないですが、散布図？

Answer 1

ラスター図がどのような図をイメージされているのかわかりませんが、散布図で描画する場合、以下のような感じでしょうか。

横軸を時間軸、縦軸を各ニューロンとし、0を白、1を黒の点で描画すればよいと思います。

import numpy as np
import matplotlib.pyplot as plt

# ダミーのデータを作成
n_neurons = 20  # ニューロン数
n_times = 100 # 時間の長さ
data = np.random.randint(0, 2, (n_times, n_neurons))
print(data)

# 描画
fig, ax = plt.subplots()

x_pos, y_pos = np.indices(data.shape)
# 1は黒、0は白の点として描画する。
ax.scatter(x_pos, y_pos, c=data, s=5, cmap="gray_r", marker='s')
ax.set_xlabel("time")
ax.set_ylabel("neurons")

plt.show()

Comments

[ryu0311]

perfectです。thx