python数组展示_python – 使用Glumpy显示NumPy数组不断更新图像

Glumpy文档是相当不存在的！这是一个简单模拟的例子,将数组可视化与glumpy与matplotlib进行比较：

import numpy as np

import glumpy

from OpenGL import GLUT as glut

from time import time

from matplotlib.pyplot import subplots,close

from matplotlib import cm

def randomwalk(dims=(256,256),n=3,sigma=10,alpha=0.95,seed=1):

""" A simple random walk with memory """

M = np.zeros(dims,dtype=np.float32)

r,c = dims

gen = np.random.RandomState(seed)

pos = gen.rand(2,n)*((r,),(c,))

old_delta = gen.randn(2,n)*sigma

while 1:

delta = (1.-alpha)*gen.randn(2,n)*sigma + alpha*old_delta

pos += delta

for ri,ci in pos.T:

if not (0. <= ri < r) : ri = abs(ri % r)

if not (0. <= ci < c) : ci = abs(ci % c)

M[ri,ci] += 1

old_delta = delta

yield M

def mplrun(niter=1000):

""" Visualise the simulation using matplotlib, using blit for

improved speed"""

fig,ax = subplots(1,1)

rw = randomwalk()

im = ax.imshow(rw.next(),interpolation='nearest',cmap=cm.hot,animated=True)

fig.canvas.draw()

background = fig.canvas.copy_from_bbox(ax.bbox) # cache the background

tic = time()

for ii in xrange(niter):

im.set_data(rw.next()) # update the image data

fig.canvas.restore_region(background) # restore background

ax.draw_artist(im) # redraw the image

fig.canvas.blit(ax.bbox) # redraw the axes rectangle

close(fig)

print "Matplotlib average FPS: %.2f" %(niter/(time()-tic))

def gprun(niter=1000):

""" Visualise the same simulation using Glumpy """

rw = randomwalk()

M = rw.next()

# create a glumpy figure

fig = glumpy.figure((512,512))

# the Image.data attribute is a referenced copy of M - when M

# changes, the image data also gets updated

im = glumpy.image.Image(M,colormap=glumpy.colormap.Hot)

@fig.event

def on_draw():

""" called in the simulation loop, and also when the

figure is resized """

fig.clear()

im.update()

im.draw( x=0, y=0, z=0, width=fig.width, height=fig.height )

tic = time()

for ii in xrange(niter):

M = rw.next() # update the array

glut.glutMainLoopEvent() # dispatch queued window events

on_draw() # update the image in the back buffer

glut.glutSwapBuffers() # swap the buffers so image is displayed

fig.window.hide()

print "Glumpy average FPS: %.2f" %(niter/(time()-tic))

if __name__ == "__main__":

mplrun()

gprun()

使用matplotlib和GTKAgg作为我的后端并使用blit来避免每次绘制背景,我可以达到大约95 FPS.使用Glumpy我得到大约250-300 FPS,即使我目前在我的笔记本电脑上设置了相当糟糕的图形设置.话虽如此,Glumpy开始工作有点繁琐,除非你处理巨大的矩阵,或者你需要一个非常高的帧率,无论出于何种原因,我都会坚持使用matplotlib和blit.