def _generate_spectra(frequencies, testdir, cc_generator): if os.path.isdir(testdir): shutil.rmtree(testdir) os.makedirs(testdir) fin = np.hstack((frequencies, frequencies)) for nr, cc_set in enumerate(cc_generator()): outdir = testdir + os.sep + '{0:03}_'.format(nr) + \ '{0}_{1}_{2}_{3}'.format(*cc_set) + os.sep os.makedirs(outdir) # generate data rlogmag_rpha = colecole.cole_log(fin, cc_set) data = np.vstack((np.exp(rlogmag_rpha[0, :]), rlogmag_rpha[1, :])).flatten() data_2d = np.atleast_2d(data) # save data np.savetxt(outdir + 'frequencies.dat', frequencies) np.savetxt(outdir + 'data.dat', data_2d) cc_set_2d = np.atleast_2d(cc_set) np.savetxt(outdir + 'colecole_orig.dat', cc_set_2d)
#!/usr/bin/python """ Generate sample one-term Cole-Cole spectrum Save to frequencies.dat and data.dat """ import numpy as np import lib_cc_fit.colecole as CC frequencies = np.logspace(-3, 3, 25) fin = np.hstack((frequencies, frequencies)) # rho0 m1 tau1 c1 m2 tau2 c2 cc_pars = [np.log(50), 0.05, np.log(40), 0.6] magpha = CC.cole_log(fin, cc_pars).flatten()[np.newaxis, :] magpha[0, 0:magpha.size / 2] = np.exp(magpha[0, 0:magpha.size / 2]) np.savetxt('data.dat', magpha) np.savetxt('frequencies.dat', frequencies)