Python VTK：直接坐标到PolyD

Question

我想把x，y和z在特定范围内的所有坐标组合直接转换为vtk.polyData公司或者vtk点. 我的第一个方法是使用itertools.product，但我认为这会有一个非常糟糕的运行时。因此，我使用vtk找到了另一种方法，不管怎样，我在下一部分的程序中都需要它。在

第一次进近itertools.product

import numpy as np
import itertools
import vtk

x1=[10,11,12....310]
y1=[10,11,12....310]
z1=[0,1,2....65]

points1 = vtk.vtkPoints()                      
for coords in itertools.product(x1,y1,z1):
   points1.InsertNextPoint(coords)
boxPolyData1 = vtk.vtkPolyData()
boxPolyData1.SetPoints(points1)

我目前对vtk的方法：

^{pr2}$

但他把我的Python撞倒了。有人有主意吗？在

谨致问候！在

Answer 1

将这些coords与^{}或^{}堆叠在列中，然后将它们作为输入馈送给num_array，如下-

x,y,z = np.mgrid[10:310, 10:310, 0:65]
out_data = np.column_stack((x.ravel(), y.ravel(), z.ravel()))

vtk_data_array = numpy_support.numpy_to_vtk(num_array=out_data,\
                              deep=True,array_type=vtk.VTK_FLOAT)

或者，要直接获得out_data，请-

^{pr2}$

另一种使用initialization替换np.mgrid创建的3D数组的方法如下-

def create_mgrid_array(d00,d01,d10,d11,d20,d21,dtype=int):
    df0 = d01-d00
    df1 = d11-d10
    df2 = d21-d20
    a = np.zeros((df0,df1,df2,3),dtype=dtype)
    X,Y,Z = np.ogrid[d00:d01,d10:d11,d20:d21]
    a[:,:,:,2] = Z
    a[:,:,:,1] = Y
    a[:,:,:,0] = X
    a.shape = (-1,3)
    return a

演示create_mgrid_array-

In [151]: create_mgrid_array(3,6,10,14,20,22,dtype=int)
Out[151]: 
array([[ 3, 10, 20],
       [ 3, 10, 21],
       [ 3, 11, 20],
       [ 3, 11, 21],
       [ 3, 12, 20],
       [ 3, 12, 21],
       [ 3, 13, 20],
       [ 3, 13, 21],
       [ 4, 10, 20],
       [ 4, 10, 21],
       [ 4, 11, 20],
       [ 4, 11, 21],
       [ 4, 12, 20],
       [ 4, 12, 21],
       [ 4, 13, 20],
       [ 4, 13, 21],
       [ 5, 10, 20],
       [ 5, 10, 21],
       [ 5, 11, 20],
       [ 5, 11, 21],
       [ 5, 12, 20],
       [ 5, 12, 21],
       [ 5, 13, 20],
       [ 5, 13, 21]])

运行时测试

方法-

def loopy_app():
    x1 = range(10,311)
    y1 = range(10,311)
    z1 = range(0,66)

    points1 = vtk.vtkPoints()                      
    for coords in itertools.product(x1,y1,z1):
       points1.InsertNextPoint(coords)
    return points1

def vectorized_app():
    out_data = create_mgrid_array(10,311,10,311,0,66,dtype=float)
    vtk_data_array = numpy_support.numpy_to_vtk(num_array=out_data,\
                                    deep=True,array_type=vtk.VTK_FLOAT)

    points2 = vtk.vtkPoints()
    points2.SetData(vtk_data_array)
    return points2

时间安排和验证-

In [155]: # Verify outputs with loopy and vectorized approaches    
     ...: out1 =  vtk_to_numpy(loopy_app().GetData())
     ...: out2 =  vtk_to_numpy(vectorized_app().GetData())
     ...: print np.allclose(out1, out2)
     ...: 
True

In [156]: %timeit loopy_app()
1 loops, best of 3: 923 ms per loop

In [157]: %timeit vectorized_app()
10 loops, best of 3: 67.3 ms per loop

In [158]: 923/67.3
Out[158]: 13.714710252600298

13x+在循环的基础上，用所提出的矢量化方法来加速！在