def plotSingleCLass(ras, decs, tag, mainClass, count): CommonPlot.plotGrid(plt) plt.title(tag + " - " + mainClass + "_Density") if count <= 5: plt.plot(ras, decs, '.') else: xy = np.vstack([ras, decs]) z = gaussian_kde(xy)(xy) # fig, ax = plt.subplots() # ax.scatter(ras, decs, c=z, s=10, edgecolors='') plt.scatter(ras, decs, c=z, s=10, edgecolors='') CommonPlot.plotGalaxy(plt) filePath = "/media/zdwdong/my/fits/Density/Density_%s_%s.png" % (tag, mainClass) plt.savefig(filePath) print tag + " : " + mainClass + " saved" plt.close()
def plotOver(lamostTable, sdssTable, mu): dic_lamost = getCoordinateDic(lamostTable) dic_sdss = getCoordinateDic(sdssTable) dic_over = {} for ra in dic_lamost.keys(): for dec in dic_lamost[ra].keys(): lcount = dic_lamost[ra][dec] scount = util.getDic2Value(dic_sdss, ra, dec) if scount is None: util.setItemToDic2(dic_over, ra, dec, 10) else: util.setItemToDic2(dic_over, ra, dec, float(lcount) / float(scount)) for ra in dic_sdss.keys(): for dec in dic_sdss[ra].keys(): if not util.isInDic2(dic_lamost, ra, dec): util.setItemToDic2(dic_over, ra, dec, 11) for ra in dic_over.keys(): for dec in dic_over[ra].keys(): alpha = dic_over[ra][dec] if alpha == 10: plotUtil.plotRectangleWithParam(plt, ra, dec, 1, 1, "red") elif alpha == 11: plotUtil.plotRectangleWithParam(plt, ra, dec, 1, 1, "blue") else: newAlpha = alpha * mu if newAlpha > 1: newAlpha = 1 plotUtil.plotRectangleWithParam(plt, ra, dec, 1, 1, "black", newAlpha) plt.title("Lamost/SDSS Covered") CommonPlot.plotGrid(plt) CommonPlot.plotGalaxy(plt) plt.savefig('/media/zdwdong/my/fits/Cover/Lamost_OVER_SDSS_' + str(mu) + '.png') plt.close()
def plotScatter(sql_distinct, sql_select_coordinate, sql_select_coordinate_count, tag, onlyShowAll=False): mainClassList = getMainClassList(sql_distinct, onlyShowAll) for mainClass in mainClassList: ras, decs, count = getCoordinateData(sql_select_coordinate, sql_select_coordinate_count, mainClass) try: CommonPlot.plotGrid(plt) plt.title(tag + " - " + mainClass + "_Scatter") plt.plot(ras, decs, '.') CommonPlot.plotGalaxy(plt) filePath = "/media/zdwdong/my/fits/Scatter/Scatter_%s_%s.png" % (tag, mainClass) plt.savefig(filePath) print tag + " : " + mainClass + " saved" plt.close() except Exception, ex: exUtil.printMessage(ex, "Density Plot Error", mainClass)