numpy中的3D数组的2D切片系列
我有一个3D数组,代表笛卡尔空间的密度值。 要获得2D图像,我只需使用sum(array,2)
求和其中的一个轴,然后使用matplotlib函数imshow(array2D)
来获得2D图像。
我想要做的就是使用imshow()
来一次只显示3D数组的一个切片,这样我就可以通过3D数组“页面”查看图像的不同点。
slice命令很简单: array[:,:,x]
但是我看不出每次至少要显示一个slice。 除了每次手动更改程序文件外,有没有人有建议? 这可以通过交互方式完成吗?
我实际上编写了代码来完成我认为你正在寻找的内容,看看是否有帮助:
import numpy as np
import pylab
class plotter:
def __init__(self, im, i=0):
self.im = im
self.i = i
self.vmin = im.min()
self.vmax = im.max()
self.fig = pylab.figure()
pylab.gray()
self.ax = self.fig.add_subplot(111)
self.draw()
self.fig.canvas.mpl_connect('key_press_event',self)
def draw(self):
if self.im.ndim is 2:
im = self.im
if self.im.ndim is 3:
im = self.im[...,self.i]
self.ax.set_title('image {0}'.format(self.i))
pylab.show()
self.ax.imshow(im, vmin=self.vmin, vmax=self.vmax, interpolation=None)
def __call__(self, event):
old_i = self.i
if event.key=='right':
self.i = min(self.im.shape[2]-1, self.i+1)
elif event.key == 'left':
self.i = max(0, self.i-1)
if old_i != self.i:
self.draw()
self.fig.canvas.draw()
def slice_show(im, i=0):
plotter(im, i)
只需调用你的3d阵列上的显示功能,我会告诉它显示哪个片段。 只要您选择了情节,您就可以用箭头键逐步切片。
请注意,这需要具有形状(x,y,z)的数组,例如,您可以使用np.dstack((im1,im2,...))从一系列2d数组中获取这样的数组。
另请参见使用两个滑块交互式绘制matplotlib,以获取使用gui滑块执行此操作的代码示例
链接地址: http://www.djcxy.com/p/8475.html上一篇: 2D slice series of 3D array in numpy
下一篇: How do I hide certain elements on my page, based on referral traffic?