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)
