import quantities as pq
from elephant import statistics, kernels
import matplotlib.pyplot as plt
from elephant.spike_train_generation import homogeneous_poisson_process
from viziphant.statistics import plot_instantaneous_rates_colormesh
np.random.seed(6)
spiketrains = [homogeneous_poisson_process(rate=10 * pq.Hz,
               t_stop=10 * pq.s) for _ in range(10)]
kernel = kernels.GaussianKernel(sigma=100 * pq.ms)
rates = statistics.instantaneous_rate(spiketrains,
                                      sampling_period=10 * pq.ms,
                                      kernel=kernel)
plot_instantaneous_rates_colormesh(rates)
plt.show()