# interval (ns) interval = float(sys.argv[2]) * float(sys.argv[3]) * 10**(-6) time = [] for i in range(len(data3)): time.append(interval * i) ## block average ------------------------------------------------- if len(sys.argv) == 5: blockSize1 = int(sys.argv[4]) blockSize2 = int(sys.argv[4]) time_ba1 = sb.blockReduceFB(time,blockSize1) area_ba1 = sb.blockReduceFB(data3[:,0],blockSize1) time_ba2 = sb.blockReduceFB(time,blockSize2) Mg_ba2 = sb.blockReduceFB(data3[:,1],blockSize2) elif len(sys.argv) == 4: acor_result1 = acor.acor(data3[:,0]) blockSize1 = int(math.ceil(acor_result1[0])) acor_result2 = acor.acor(data3[:,1]) blockSize2 = int(math.ceil(acor_result2[0])) time_ba1 = sb.blockReduceFB(time,blockSize1) area_ba1 = sb.blockReduceFB(data3[:,0],blockSize1) time_ba2 = sb.blockReduceFB(time,blockSize2) Mg_ba2 = sb.blockReduceFB(data3[:,1],blockSize2)
data2 = np.loadtxt(name2) data3 = np.loadtxt(name3) data4 = np.loadtxt(name4) ## Set Output Name ------------------------------------------------- blockSize = int(sys.argv[2]) s='_area_bulkIon_b%d.pdf' % (blockSize) outputName = sys.argv[1]+s ## block average ------------------------------------------------- area_ba = sb.blockReduceFB(data1[:,0],blockSize) P_ba = sb.blockReduceFB(data1[:,1],blockSize) Cl_ba = sb.blockReduceFB(data2[:,1],blockSize) Mg_ba = sb.blockReduceFB(data3[:,1],blockSize) K_ba = sb.blockReduceFB(data4[:,1],blockSize) ## Pearson Correlation -------------------------------------------- #print scipy.stats.pearsonr(area_ba,P_ba) Cl_r, Cl_p = scipy.stats.pearsonr(area_ba,Cl_ba) Mg_r, Mg_p = scipy.stats.pearsonr(area_ba,Mg_ba) K_r, K_p = scipy.stats.pearsonr(area_ba,K_ba) ## Plotting -------------------------------------------------------
## Load Input File ------------------------------------------------- data0 = np.loadtxt('../Dropbox/Meeting_ppt/2014_6_2_meeting/square2plate_1MKCl_remove_Mg_npt40_Mg_area_bulkIon.dat') data1 = np.loadtxt('../Dropbox/Meeting_ppt/2014_6_2_meeting/square2plate_1MKCl_reduce_Mg3_npt_extend25_mghh25_Mg_area_bulkIon.dat') data2 = np.loadtxt('../Dropbox/Meeting_ppt/2014_6_2_meeting/square2plate_1MKCl_60_Mg_npt16_mghh25_Mg_area_bulkIon.dat') data3 = np.loadtxt('../Dropbox/Meeting_ppt/2014_6_2_meeting/square2plate_1MKCl_73_Mg_npt15_mghh25_Mg_area_bulkIon.dat') data4 = np.loadtxt('../Dropbox/Meeting_ppt/2014_6_2_meeting/square2plate_1MKCl_86_Mg_npt17_mghh25_Mg_area_bulkIon.dat') data5 = np.loadtxt('../Dropbox/Meeting_ppt/2014_6_2_meeting/square2plate_1MKCl_99_Mg_npt17_mghh25_Mg_area_bulkIon.dat') data6 = np.loadtxt('../Dropbox/Meeting_ppt/2014_6_2_meeting/square2plate_1MKCl_112_Mg_npt17_mghh25_Mg_area_bulkIon.dat') data7 = np.loadtxt('../Dropbox/Meeting_ppt/2014_6_2_meeting/square2plate_1MKCl_npt_extend38_mghh25_Mg_area_bulkIon.dat') ## block average ------------------------------------------------- acor_result0 = acor.acor(data0[:,0]) blockSize0 = int(math.ceil(acor_result0[0])) area0_ba = sb.blockReduceFB(data0[:,0],blockSize0) Mg0_ba = sb.blockReduceFB(data0[:,1],blockSize0) acor_result1 = acor.acor(data1[:,0]) blockSize1 = int(math.ceil(acor_result1[0])) area1_ba = sb.blockReduceFB(data1[:,0],blockSize1) Mg1_ba = sb.blockReduceFB(data1[:,1],blockSize1) acor_result2 = acor.acor(data2[:,0]) blockSize2 = int(math.ceil(acor_result2[0])) area2_ba = sb.blockReduceFB(data2[:,0],blockSize2) Mg2_ba = sb.blockReduceFB(data2[:,1],blockSize2) acor_result3 = acor.acor(data3[:,0]) blockSize3 = int(math.ceil(acor_result3[0])) area3_ba = sb.blockReduceFB(data3[:,0],blockSize3)
elif dim == 'z': area = [] for i in range(len(array)): area.append(array[i][1] * array[i][2]) ## block average ---------------------------------------------- if len(sys.argv) == 6: blockSize = int(sys.argv[5]) elif len(sys.argv) == 5: acor_result = acor.acor(area) blockSize = int(math.ceil(acor_result[0])) time_ba = sb.blockReduceFB(time,blockSize) area_ba = sb.blockReduceFB(area,blockSize) ## set output file name -------------------------------------- inputPrefix = re.split('\.', sys.argv[1]) inputPrefix_noType = '' for i in range(len(inputPrefix)-1): inputPrefix_noType = inputPrefix_noType + inputPrefix[i] if i < (len(inputPrefix)-2): inputPrefix_noType = inputPrefix_noType + '.' s='_area_b%d.pdf' % (blockSize) outputName = inputPrefix_noType+s