Skip to content

Commit

Permalink
Cosmetic updates to plots
Browse files Browse the repository at this point in the history
  • Loading branch information
@ycanerol
ycanerol committed Jun 28, 2017
1 parent ce9f5b2 commit 13cc483
Show file tree
Hide file tree
Showing 2 changed files with 40 additions and 30 deletions.
43 changes: 25 additions & 18 deletions figure_plotter.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,11 +14,11 @@


# Exp date Cluster Label Flipped
plotthis = [['2017_02_14', '901', 'On cell', False],
['2017_01_31', '8401', 'Off cell', True],
['2017_01_17', '201', 'On off 1', True],
['2017_01_17', '4105', 'Off to on',False],
['2017_02_14', '22302', 'On to off',True],
plotthis = [['2017_02_14', '901', 'on_cell', False],
['2017_01_31', '8401', 'off_cell', True],
['2017_01_17', '201', 'onoff1', True],
['2017_01_17', '4105', 'off_to_on',False],
['2017_02_14', '22302', 'on_to_off',True],
]


Expand Down Expand Up @@ -49,14 +49,14 @@
matplotlib.rcParams['axes.spines.bottom'] = True

plt.figure(figsize=(12, 12), dpi=200)
plt.suptitle(str(' '.join(str(c['spike_path'])
.split('rasters')[0].split('Experiments')[1]
.split('/'))+str(i[1])))
# plt.suptitle(str(' '.join(str(c['spike_path'])
# .split('rasters')[0].split('Experiments')[1]
# .split('/'))+str(i[1])))
plt.subplot(3, 3, 1)
plt.plot(f['sta']*flipper)
plt.plot(f['v'][:, 0]*flipper)
plt.title('Filters')
plt.legend(['STA', 'Eigenvalue 0'], fontsize='small')
plt.legend(['STA', 'STC'], fontsize='small')
plt.xticks(np.linspace(0, 20, 3), np.linspace(0, 600, 3).astype(int))
plt.ylabel('Full field flicker\n$\\regular_{Linear\,\,\,output}$',
fontsize=16)
Expand All @@ -65,8 +65,8 @@
ax = plt.subplot(3, 3, 2)
plt.plot(f['bins_sta'], f['spikecount_sta'][::flipper], '-')
plt.plot(f['bins_stc'], f['spikecount_stc'][::flipper], '-')
plt.text(.5, .99, 'On-Off Bias: {:2.2f}\nTotal spikes: {}'
.format(float(f['onoffindex']), f['total_spikes']),
plt.text(.5, .99, 'On-Off index: {:2.2f}'
.format(float(f['onoffindex'])),
horizontalalignment='center',
verticalalignment='top',
transform=ax.transAxes)
Expand All @@ -89,14 +89,14 @@
plt.ylabel('Checkerflicker\n$\\regular_{Linear\,\,\,output}$', fontsize=16)
plt.xlabel('Time [ms]')
plt.xticks(np.linspace(0, 20, 3), np.linspace(0, 600, 3).astype(int))
plt.legend(['Weighted stimulus', 'Eigenvalue 0', 'Brightest pixel'],
plt.legend(['STA', 'STC', 'Center pixel'],
fontsize='small')

ax = plt.subplot(3, 3, 5)
for j in range(len(c['bins'])):
plt.plot(c['bins'][j], c['spike_counts_in_bins'][j][::flipper], '-')
plt.text(.5, .99, 'On-Off Bias: {:2.2f}\nTotal spikes: {}'
.format(float(c['onoffindex']), c['total_spikes']),
plt.text(.5, .99, 'On-Off index: {:2.2f}'
.format(float(c['onoffindex'])),
horizontalalignment='center',
verticalalignment='top',
transform=ax.transAxes)
Expand All @@ -123,16 +123,23 @@
vmax=np.max(c['sta_unscaled']))
plt.title('Receptive field')

plt.subplot(3, 3, 8)
ax = plt.subplot(3, 3, 8)
f_size = 5
plt.imshow(c['sta_unscaled'][c['max_i'][0]-f_size:c['max_i'][0]+f_size+1,
c['max_i'][1]-f_size:c['max_i'][1]+f_size+1,
int(c['max_i'][2])],
cmap='Greys',
vmin=np.min(c['sta_unscaled']),
vmax=np.max(c['sta_unscaled']))
plt.title('Brightest pixel: {}'.format(c['max_i']))
vmax=np.max(c['sta_unscaled']),
extent=[-375, 375, -375, 375])
ax.add_patch(matplotlib.patches.Rectangle(
(-37.5, -37.5), 75, 75, linewidth=3,
edgecolor='C2', facecolor='none'))
plt.title('Center pixel')
plt.xticks(np.arange(-300, 301, 100))
plt.xlabel('Distance [µm]')
plt.ylabel('Distance [µm]')
plt.tight_layout(pad=5, h_pad=1, w_pad=1.8)
# plt.show()
plt.savefig(savepath, dpi=200, bbox_inches='tight')
# plt.show()
plt.close()
27 changes: 15 additions & 12 deletions onoffplotter_dimos.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -209,6 +209,7 @@
exp_name = 'all_experiments'

hist_axis_limits = [-1, 1, 0, 30]
r_fontsize=14

w_pf = np.sum(dataset_sizes[:2]) # Index at which data with preframes start

Expand All @@ -229,9 +230,10 @@
r_co_npf = np.corrcoef(o_npf, c_npf)[1, 0]

plt.figure(figsize=(10, 10), dpi=200)
plt.suptitle(exp_name)
#plt.suptitle(exp_name)

plt.subplot(6, 2, 1)
plt.title('Distribution of on-off indices')
plt.hist(all_f, bins=np.linspace(-1, 1, num=40),
alpha=.5, color='C0')
plt.xticks(ticks)
Expand All @@ -255,7 +257,8 @@
plt.subplot(2, 2, 2)
plt.plot(np.linspace(-1, 1), np.linspace(-1, 1), '--', alpha=.4, color='black')
plt.plot(all_f, all_c, '.', alpha=.5)
plt.text(.7, -1.05, 'r = {:4.2}'.format(r_cf))
plt.text(1, -1.05, 'r = {:4.2}'.format(r_cf), fontsize=r_fontsize,
horizontalalignment='right')
plt.axis(axis_limits)
plt.axis('square')
plt.xticks(ticks)
Expand All @@ -269,12 +272,12 @@
plt.plot(np.linspace(-1, 1), np.linspace(-1, 1), '--', alpha=.4, color='black')
plt.plot(f_npf, o_npf, '.', alpha=.5)
plt.plot(f_pf, o_pf, '.', alpha=.5)
plt.text(1, -0.85, r'$r_{{total}}$ = {:4.2}'.format(r_of),
horizontalalignment='right')
plt.text(1, -0.95, r'$r_{{no\;preframes}}$ = {:4.2}'.format(r_of_npf),
color='C0', horizontalalignment='right')
plt.text(1, -0.81, r'$r_{{total}}$ = {:4.2}'.format(r_of),
fontsize=r_fontsize, horizontalalignment='right')
plt.text(1, -0.93, r'$r_{{no\;preframes}}$ = {:4.2}'.format(r_of_npf),
fontsize=r_fontsize, color='C0', horizontalalignment='right')
plt.text(1, -1.05, r'$r_{{with\;preframes}}$ = {:4.2}'.format(r_of_pf),
color='C1', horizontalalignment='right')
fontsize=r_fontsize, color='C1', horizontalalignment='right')
plt.axis(axis_limits)
plt.axis('square')
plt.xticks(ticks)
Expand All @@ -288,12 +291,12 @@
plt.plot(np.linspace(-1, 1), np.linspace(-1, 1), '--', alpha=.4, color='black')
plt.plot(c_npf, o_npf, '.', alpha=.5)
plt.plot(c_pf, o_pf, '.', alpha=.5)
plt.text(1, -0.85, r'$r_{{total}}$ = {:4.2}'.format(r_co),
horizontalalignment='right')
plt.text(1, -0.95, r'$r_{{no\;preframes}}$ = {:4.2}'.format(r_co_npf),
color='C0', horizontalalignment='right')
plt.text(1, -0.81, r'$r_{{total}}$ = {:4.2}'.format(r_co),
fontsize=r_fontsize, horizontalalignment='right')
plt.text(1, -0.93, r'$r_{{no\;preframes}}$ = {:4.2}'.format(r_co_npf),
fontsize=r_fontsize, color='C0', horizontalalignment='right')
plt.text(1, -1.05, r'$r_{{with\;preframes}}$ = {:4.2}'.format(r_co_pf),
color='C1', horizontalalignment='right')
fontsize=r_fontsize, color='C1', horizontalalignment='right')
plt.axis(axis_limits)
plt.axis('square')
plt.xticks(ticks)
Expand Down

0 comments on commit 13cc483

Please sign in to comment.