将matplotlib hexbin放入Aitoff投影中

时间:2017-09-20 11:22:16

标签: python matplotlib map-projections astropy hexagonal-tiles

下面我有一个很好的hexbin图,但我想知道是否有办法让hexbin进入Aitoff投影?突出的代码是:

import numpy as np
import math
import matplotlib.pyplot as plt
from astropy.io import ascii

filename = 'WISE_W4SNRge3_and_W4MPRO_lt_6.0_RADecl_nohdr.dat'
datafile= path+filename
data = ascii.read(datafile)  
points = np.array([data['ra'], data['dec']])

color_map = plt.cm.Spectral_r 
points = np.array([data['ra'], data['dec']]) 
xbnds = np.array([ 0.0,360.0]) 
ybnds = np.array([-90.0,90.0]) 
extent = [xbnds[0],xbnds[1],ybnds[0],ybnds[1]] 

fig = plt.figure(figsize=(6, 4)) 
ax = fig.add_subplot(111) 
x, y = points 
gsize = 45 
image = plt.hexbin(x,y,cmap=color_map, 
    gridsize=gsize,extent=extent,mincnt=1,bins='log') 

counts = image.get_array() 
ncnts = np.count_nonzero(np.power(10,counts)) 
verts = image.get_offsets() 

ax.set_xlim(xbnds) 
ax.set_ylim(ybnds) 
plt.xlabel('R.A.')  
plt.ylabel(r'Decl.') 
plt.grid(True) 
cb = plt.colorbar(image, spacing='uniform', extend='max') 
plt.show()

我试过了:

plt.subplot(111, projection="aitoff")

在执行plt.hexbin命令之前,但只提供一个漂亮但空白的Aitoff网格。

enter image description here

2 个答案:

答案 0 :(得分:4)

问题是Aitoff投影使用弧度,从-π到+π。不是从0到360的度数。我使用Angle.wrap_at函数来实现这一点,根据this Astropy example(它基本上告诉你如何创建一个正确的Aitoff投影图)。

此外,您无法更改轴限制(这将导致错误),并且不应使用extent(因为ImportanceOfBeingErnest的答案也会说明)。

您可以按照以下方式更改代码,以获得所需内容:

import numpy as np
import matplotlib.pyplot as plt
from astropy.io import ascii
from astropy.coordinates import SkyCoord
from astropy import units

filename = 'WISE_W4SNRge3_and_W4MPRO_lt_6.0_RADecl_nohdr.dat'
data = ascii.read(filename)
coords = SkyCoord(ra=data['ra'], dec=data['dec'], unit='degree')
ra = coords.ra.wrap_at(180 * units.deg).radian
dec = coords.dec.radian

color_map = plt.cm.Spectral_r
fig = plt.figure(figsize=(6, 4))
fig.add_subplot(111, projection='aitoff')
image = plt.hexbin(ra, dec, cmap=color_map,
                   gridsize=45, mincnt=1, bins='log')

plt.xlabel('R.A.')
plt.ylabel('Decl.')
plt.grid(True)
plt.colorbar(image, spacing='uniform', extend='max')
plt.show()

哪个给出 enter image description here

答案 1 :(得分:1)

我猜你的问题在于extent的使用,它被设置为球面坐标系范围以外的其他东西。

以下工作正常:

import matplotlib.pyplot as plt
import numpy as np

ra = np.linspace(-np.pi/2.,np.pi/2.,1000)
dec = np.sin(ra)*np.pi/2./2.
points = np.array([ra, dec]) 

plt.subplot(111, projection="aitoff")

color_map = plt.cm.Spectral_r 
x, y = points 
gsize = 45 
image = plt.hexbin(x,y,cmap=color_map, 
                   gridsize=45,mincnt=1,bins='log') 

plt.xlabel('R.A.')  
plt.ylabel(r'Decl.') 
plt.grid(True) 
cb = plt.colorbar(image, spacing='uniform', extend='max') 
plt.show()

enter image description here