Merge pull request #474 from petermarathas/fix_plot_spike_ylabels

Hananel-Hazan · web-flow · commit d3eca0d1f91f · 2021-04-15T00:13:33.000Z
fix yticks in plot_spikes
diff --git a/bindsnet/analysis/plotting.py b/bindsnet/analysis/plotting.py
@@ -140,13 +140,13 @@ def plot_spikes(
             axes[i].set_title(
                 "%s spikes for neurons (%d - %d) from t = %d to %d " % args
             )
+            axes[i].set_yticks([n_neurons[datum[0]][0], n_neurons[datum[0]][1]])
         for ax in axes:
             ax.set_aspect("auto")
 
         plt.setp(
             axes,
             xticks=[],
-            yticks=[],
             xlabel="Simulation time",
             ylabel="Neuron index",
         )
@@ -174,6 +174,7 @@ def plot_spikes(
             axes[i].set_title(
                 "%s spikes for neurons (%d - %d) from t = %d to %d " % args
             )
+            axes[i].set_yticks([n_neurons[datum[0]][0], n_neurons[datum[0]][1]])
 
     plt.draw()
 

Original file line number	Diff line number	Diff line change
`@@ -140,13 +140,13 @@ def plot_spikes(`
`140`	`140`	`axes[i].set_title(`
`141`	`141`	`"%s spikes for neurons (%d - %d) from t = %d to %d " % args`
`142`	`142`	`)`
	`143`	`+ axes[i].set_yticks([n_neurons[datum[0]][0], n_neurons[datum[0]][1]])`
`143`	`144`	`for ax in axes:`
`144`	`145`	`ax.set_aspect("auto")`
`145`	`146`
`146`	`147`	`plt.setp(`
`147`	`148`	`axes,`
`148`	`149`	`xticks=[],`
`149`		`- yticks=[],`
`150`	`150`	`xlabel="Simulation time",`
`151`	`151`	`ylabel="Neuron index",`
`152`	`152`	`)`
`@@ -174,6 +174,7 @@ def plot_spikes(`
`174`	`174`	`axes[i].set_title(`
`175`	`175`	`"%s spikes for neurons (%d - %d) from t = %d to %d " % args`
`176`	`176`	`)`
	`177`	`+ axes[i].set_yticks([n_neurons[datum[0]][0], n_neurons[datum[0]][1]])`
`177`	`178`
`178`	`179`	`plt.draw()`
`179`	`180`