imshow和histogram2d:无法让他们工作

时间:2012-07-06 18:24:32

标签: python numpy matplotlib

我正在学习Python,这是我的第一个问题。我已经阅读了与 imshow 的使用相关的其他主题,但没有找到任何有用的内容。抱歉我的英语不好。

我在这里绘制了一组点,左图:

points (left) and image (right)

现在我想看一个点密度的图像,所以我使用了 imshow histogram2d ,我在前面的图像中找到了链接。

图像与点的分布不对应。这怎么可能?我已按照帮助中的说明进行操作,甚至更改了一些参数,但没有任何效果:(

代码是:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm

j, h, k = np.loadtxt("test.dat", usecols=(2, 4, 6), \
    unpack=True)

# límites
xmin = -0.5
xmax =  3.0
ymin = -0.5
ymax =  4.0

# colores
j_h = j - h
h_k = h - k

# no todas las estrellas son graficadas    
x1 = 0.5
y1 = 0.5
b  = 2.2
c  = y1 - b * x1

x = y = np.array([])

for xi, yi in zip(h_k, j_h):
    if xi < (yi - c) / b:
        x = np.append(x, xi)
        y = np.append(y, yi)

# gráfico
fig = plt.figure(figsize=(8, 7))

ax = fig.add_subplot(111)
#ax.plot(x, y, "go")
ax.set_xlabel(r"X", fontsize=14)
ax.set_ylabel(r"Y", fontsize=14)
ax.axis([xmin, xmax, ymin, ymax])

# imagen
rango = [[xmin, xmax], [ymin, ymax]]
binsx = int((xmax - xmin) / 0.05)
binsy = int((ymax - ymin) / 0.05)
binsxy = [binsx, binsy]

H, xedges, yedges = np.histogram2d(x, y, range=rango, bins=binsxy)

extent = [yedges[0], yedges[-1], xedges[0], xedges[-1]]
cp = ax.imshow(H, interpolation='bilinear', extent=extent, cmap=cm.jet)
fig.colorbar(cp)

plt.show()

所用数据的链接如下:

https://dl.dropbox.com/u/10411539/python/test.dat

感谢任何帮助!

2 个答案:

答案 0 :(得分:5)

尝试不同的插值,并转置矩阵以使其在同一轴上:

cp = ax.imshow(H.transpose()[::-1], interpolation='nearest', extent=extent, cmap=cm.jet)

答案 1 :(得分:4)

这是你想要的吗?如果要传递x和y坐标,可以使用pcolor(和pcolormesh)。

import urllib
import numpy as np
import matplotlib.pyplot as plt
f = urllib.urlopen('https://dl.dropbox.com/u/10411539/python/test.dat')
j, h, k = np.loadtxt(f, usecols=(2, 4, 6), \
    unpack=True)
j, h, k
j_h = j - h
h_k = h - k
H, xedges, yedges = np.histogram2d(j_h, h_k, bins=100)
plt.pcolor(xedges, yedges, H)

Example of pcolor using histogram2d

对于imshow,你必须反转第一个维度,因为imshow使用从零开始的行,列索引到x,y。从上往下绘制。

plt.imshow(H[::-1,:], extent=(0,5, 0,2.5)) # typed in extent by hand.

imshow example