在python中将数组显示为光栅图像

Question

我在Python中有一个numpy数组,我想在屏幕上显示为光栅图像.最简单的方法是什么？它不需要特别花哨或具有良好的界面,我需要做的就是将数组的内容显示为灰度光栅图像.

我正在尝试使用NumPy将我的一些IDL代码转换为Python,并且我基本上正在寻找IDL中的tv和tvscl命令的替代品.

Answer 1

根据您的需要,matplotlibimshow或glumpy可能是最好的选择.

Matplotlib无限灵活,但速度较慢(即使你做的一切都正确,matplotlib中的动画也可能是令人惊讶的资源密集型).但是,您将拥有一个非常精彩,功能齐全的绘图库供您使用.

Glumpy非常适合基于OpenGL的快速显示和2D numpy阵列动画,但它的功能更加有限.如果你需要实时制作一系列图像或显示数据的动画,那么它比matplotlib要好得多.

使用matplotlib(使用pyplot API而不是pylab):

import matplotlib.pyplot as plt
import numpy as np

# Generate some data...
x, y = np.meshgrid(np.linspace(-2,2,200), np.linspace(-2,2,200))
x, y = x - x.mean(), y - y.mean()
z = x * np.exp(-x**2 - y**2)

# Plot the grid
plt.imshow(z)
plt.gray()
plt.show()

使用glumpy:

import glumpy
import numpy as np

# Generate some data...
x, y = np.meshgrid(np.linspace(-2,2,200), np.linspace(-2,2,200))
x, y = x - x.mean(), y - y.mean()
z = x * np.exp(-x**2 - y**2)

window = glumpy.Window(512, 512)
im = glumpy.Image(z.astype(np.float32), cmap=glumpy.colormap.Grey)

@window.event
def on_draw():
    im.blit(0, 0, window.width, window.height)
window.mainloop()

Answer 2

在pylab交互模式下使用ipython，您可以执行以下操作：

$ ipython pylab
In [1]: imshow(your_array)

或不在pylab模式下：

$ ipython
In [1]: from pylab import *
In [2]: imshow(your_array)
In [3]: pylab.show()

或没有pylab名称空间的东西：

$ ipython
In [1]: import matplotlib.pyplot as pyplot
In [2]: pyplot.imshow(your_array)
In [3]: pyplot.show()