def specContourWithXYZout(L, inputFolder=inputFolderWRF, label="", vmin="", vmax="", title=""): if label=="": label = str(time.time()) plt.close() Z = np.zeros((13,8)) for frameCount, fileName in enumerate(L): XYZ = pickle.load(open(inputFolder+fileName,'r')) #X = XYZ['X'] #Y = XYZ['Y'] Z1 = XYZ['Z'] Z += Z1 XYZ['Z'] = Z/ (frameCount+1) #vmins = (np.log10(XYZ["Z"])* (Z>0)).min() #vmaxs = (np.log10(XYZ["Z"])* (Z>0)).max() X = XYZ['X'] Y = XYZ['Y'] XYZout = specContour.specContour(XYZ, display=True, outputFolder=outputFolder, #vmin=-1.0, vmax=3.6, vmin=vmin, vmax=vmax, fileName = testName+ label + "_average_of_" + str(frameCount+1) +'images.png', title=title, ) plt.close() print testName, "number of frames", frameCount+1 return XYZout
for count , L in enumerate([Ltotal, Lmax]): plt.close() Z = np.zeros((13,8)) for frameCount, fileName in enumerate(L): XYZ = pickle.load(open(inputFolder+fileName,'r')) #X = XYZ['X'] #Y = XYZ['Y'] Z1 = XYZ['Z'] Z += Z1 XYZ['Z'] = Z/ (frameCount+1) #vmins[count] = (np.log10(XYZ["Z"])* (Z>0)).min() #vmaxs[count] = (np.log10(XYZ["Z"])* (Z>0)).max() X = XYZ['X'] Y = XYZ['Y'] XYZouts[count] = specContour.specContour(XYZ, display=True, outputFolder=outputFolder, vmin=-1.0, vmax=3.6, fileName = testName+ labels[count] + "_average_of_" + str(frameCount+1) +'images.png') plt.close() XYZs[count] = {'X': X.copy(), 'Y': Y.copy(), 'Z': Z.copy()} specContour.specContour(XYZs[0], XYZs[1], outputFolder=outputFolder, fileName=testName+"total-max.png") #specContour.specContour(XYZs[0], XYZs[1], outputFolder=outputFolder, fileName=testName+"total-max.png", vmin=-.8, vmax=3.17) print testName, "number of frames", frameCount+1 # ## #### ######## ## set the "setMin/setMax" - edit here #vmax = max(vmaxs)