如何用 x 和 y 坐标绘制二维数据（图像）答案

【问题标题】：How to plot 2D data (image) with x and y coordinates如何用 x 和 y 坐标绘制二维数据（图像）
【发布时间】：2021-03-01 22:41:26
【问题描述】：

我想使用 pyqtgraph 绘制一个二维光谱，其中 x 坐标是光谱维度（波长），y 坐标是空间维度（以天空的弧秒为单位）。

我已经能够使用 ImageItem() 来做到这一点，但我似乎无法弄清楚如何在正确的坐标中显示 x 和 y 轴。我不想只更改标签或刻度，而是要更改绘图的坐标，因为我稍后需要使用这些值（波长和弧秒）执行操作。

这是一个最小的工作示例：

import pyqtgraph as pg
import numpy as np


# The fake data
wvlg = np.linspace(300, 600, 5000)
arcsec = np.linspace(-5, 5, 100)
flux = np.ones((wvlg.shape[0], arcsec.shape[0])) * np.exp(-(arcsec)**2/0.1)
flux += np.random.normal(0, 0.1, size=(wvlg.shape[0], arcsec.shape[0]))

# The plotting
win = pg.GraphicsLayoutWidget(show=True)
ax2D = win.addPlot(title='2D spectrum', row=0, col=0)
img = pg.ImageItem()
img.setImage(flux)
ax2D.addItem(img)
# Some line converting the x and y values to wvlg and arcsec

这给出了一个图像，其中 x 和 y 轴显示索引值，而我想显示相应的波长和 arcsec 值。

有没有一种我在文档中严重忽略的简单方法？

【问题讨论】：

标签： python pyqt5 pyqtgraph

【解决方案1】：

您可以使用ImageItem 类的setRect 方法来设置数据的范围。请参阅下面的示例。

请注意，我将图像移动了半个像素，以便像素中心与精确坐标匹配。否则坐标将与像素的角点之一对齐。

import pyqtgraph as pg
import numpy as np

from PyQt5 import QtCore, QtWidgets


def main():
    app = QtWidgets.QApplication([])
    
    # The fake data
    wvlg = np.linspace(300, 600, 5000)
    arcsec = np.linspace(-5, 5, 100)
    flux = np.ones((wvlg.shape[0], arcsec.shape[0])) * np.exp(-(arcsec)**2/0.1)
    flux += np.random.normal(0, 0.1, size=(wvlg.shape[0], arcsec.shape[0]))

    # The plotting
    win = pg.GraphicsLayoutWidget()
    ax2D = win.addPlot(title='2D spectrum', row=0, col=0)
    img = pg.ImageItem()
    img.setImage(flux)
    ax2D.addItem(img)
    
    print(flux.shape)

    # Move the image by half a pixel so that the center of the pixels are
    # located at the coordinate values
    dx = wvlg[1]-wvlg[0]
    dy = arcsec[1]-arcsec[0]
    print("pixel size x: {}, pixel size y: {}".format(dx, dy))
    
    rect = QtCore.QRectF(wvlg[0] - dx/2, arcsec[0] - dy/2, 
                         wvlg[-1] - wvlg[0], arcsec[-1] - arcsec[0])
    print(rect)
    img.setRect(rect)

    ax2D.setLabels(left='arcsec', bottom='wvlg')
    
    win.show()
    win.raise_()
    app.exec_()
    
if __name__ == "__main__":
    main()

【讨论】：