benda/nonlinearbaseline2025
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated data and figures to new analysis with the right units

Browse Source
This commit is contained in:
Jan Benda
2025-05-25 23:12:11 +02:00
parent e87d63c46b
commit d043a8c47c
423 changed files with 607 additions and 923 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
plotstyle.py
View File

@@ -33,7 +33,7 @@ def noise_files(data_path, cell_name, alpha=None):
return files, nums
def plot_chi2(ax, s, freqs, chi2, fcutoff, rate=None):
def plot_chi2(ax, s, freqs, chi2, fcutoff, rate=None, vmax=None):
ax.set_visible(True)
ax.set_aspect('equal')
i0 = np.argmin(freqs < 0)
@@ -43,31 +43,35 @@ def plot_chi2(ax, s, freqs, chi2, fcutoff, rate=None):
freqs = freqs[i0:i1]
chi2 = 1e-4*chi2[i0:i1, i0:i1] # Hz/%^2
vquantile = 0.996
vmax = np.quantile(chi2, vquantile)
if vmax is None:
vmax = np.quantile(chi2, vquantile)
ten = 10**np.floor(np.log10(vmax))
for fac, delta in zip([1, 2, 3, 4, 6, 8, 10],
[0.5, 1, 1, 2, 3, 4, 5]):
for fac, delta in zip([1, 1.2, 1.5, 2, 3, 4, 6, 8, 10],
[0.5, 0.4, 0.5, 1, 1, 2, 3, 4, 5]):
if fac*ten >= vmax:
#vmax = prev_fac*ten
#ten *= prev_delta
vmax = fac*ten
ten *= delta
break
prev_fac = fac
prev_delta = delta
#prev_fac = fac
#prev_delta = delta
pc = ax.pcolormesh(freqs, freqs, chi2, vmin=0, vmax=vmax,
rasterized=True)
ax.set_xlim(0, fcutoff)
ax.set_ylim(0, fcutoff)
ax.set_xticks_delta(100)
ax.set_yticks_delta(100)
df = 100 if fcutoff > 250 else 50
ax.set_xticks_delta(df)
ax.set_yticks_delta(df)
ax.set_xlabel('$f_1$', 'Hz')
ax.set_ylabel('$f_2$', 'Hz')
if rate is not None:
dfreqs, diag = diag_projection(freqs, chi2, 2*fcutoff)
nli, nlirel, nlif = peak_size(dfreqs, diag, rate, median=False)
ax.text(0.95, 0.88, f'SI($r$)={nli:.1f}', ha='right', zorder=50,
color='white', fontsize='medium', transform=ax.transAxes)
color='white', fontsize='medium', transform=ax.transAxes,
bbox=dict(boxstyle='round,pad=0.1', ec='none', fc='black',
alpha=0.4))
cax = ax.inset_axes([1.04, 0, 0.05, 1])
cax.set_spines_outward('lrbt', 0)
cb = ax.get_figure().colorbar(pc, cax=cax)