2

我在用着

p = VN.vtk_to_numpy(data.GetCellData().GetArray('p'))

从以这种方式编写的 .vtk 文件中读取 3D 标量:

p_x1y1z1 p_x2y1z1 p_x3y1z1 p_x4y1z1 p_x1y2z1 p_x2y2z1 p_x3y2z1 p_x4y2z1

依此类推,循环围绕x,yz.

我想用这些数据填充一个 3D numpy 数组(它是一个常规网格),p(i,j,k)=p_ijk这样我就可以使用 numpy 工具箱中的渐变和其他运算符。

有任何想法吗?

问候

4

1 回答 1

1

如果我正确理解你的情况,你就可以reshape了。

In [132]: p = np.array("p_x1y1z1 p_x2y1z1 p_x3y1z1 p_x4y1z1 p_x1y2z1 p_x2y2z1 p_x3y2z1 p_x4y2z1".split())

In [133]: p
Out[133]: 
array(['p_x1y1z1', 'p_x2y1z1', 'p_x3y1z1', 'p_x4y1z1', 'p_x1y2z1', 'p_x2y2z1', 'p_x3y2z1', 'p_x4y2z1'], 
      dtype='|S8')

在我看来,您的数组是按照 numpy 所谓的排序方式'F'排序的:

In [168]: p.reshape(4, 2, order='F')
Out[168]: 
array([['p_x1y1z1', 'p_x1y2z1'],
       ['p_x2y1z1', 'p_x2y2z1'],
       ['p_x3y1z1', 'p_x3y2z1'],
       ['p_x4y1z1', 'p_x4y2z1']], 
      dtype='|S8')

如果您也有z差异,只需重塑为三个维度:

In [169]: q
Out[169]: 
array(['p_x1y1z1', 'p_x2y1z1', 'p_x3y1z1', 'p_x4y1z1', 'p_x1y2z1',
       'p_x2y2z1', 'p_x3y2z1', 'p_x4y2z1', 'p_x1y1z2', 'p_x2y1z2',
       'p_x3y1z2', 'p_x4y1z2', 'p_x1y2z2', 'p_x2y2z2', 'p_x3y2z2',
       'p_x4y2z2', 'p_x1y1z3', 'p_x2y1z3', 'p_x3y1z3', 'p_x4y1z3',
       'p_x1y2z3', 'p_x2y2z3', 'p_x3y2z3', 'p_x4y2z3'], 
      dtype='|S8')

In [170]: q.reshape(4,2,3,order='F')
Out[170]: 
array([[['p_x1y1z1', 'p_x1y1z2', 'p_x1y1z3'],
        ['p_x1y2z1', 'p_x1y2z2', 'p_x1y2z3']],

       [['p_x2y1z1', 'p_x2y1z2', 'p_x2y1z3'],
        ['p_x2y2z1', 'p_x2y2z2', 'p_x2y2z3']],

       [['p_x3y1z1', 'p_x3y1z2', 'p_x3y1z3'],
        ['p_x3y2z1', 'p_x3y2z2', 'p_x3y2z3']],

       [['p_x4y1z1', 'p_x4y1z2', 'p_x4y1z3'],
        ['p_x4y2z1', 'p_x4y2z2', 'p_x4y2z3']]], 
      dtype='|S8')

这假设x,y,z应该映射到i+1,j+1,k+1,如下所示:

In [175]: r = q.reshape(4,2,3,order='F')

In [176]: r[0]   #all x==1
Out[176]: 
array([['p_x1y1z1', 'p_x1y1z2', 'p_x1y1z3'],
       ['p_x1y2z1', 'p_x1y2z2', 'p_x1y2z3']], 
      dtype='|S8')

In [177]: r[:,0]  # all y==1
Out[177]: 
array([['p_x1y1z1', 'p_x1y1z2', 'p_x1y1z3'],
       ['p_x2y1z1', 'p_x2y1z2', 'p_x2y1z3'],
       ['p_x3y1z1', 'p_x3y1z2', 'p_x3y1z3'],
       ['p_x4y1z1', 'p_x4y1z2', 'p_x4y1z3']], 
      dtype='|S8')

In [178]: r[:,:,0]  #all z==1
Out[178]: 
array([['p_x1y1z1', 'p_x1y2z1'],
       ['p_x2y1z1', 'p_x2y2z1'],
       ['p_x3y1z1', 'p_x3y2z1'],
       ['p_x4y1z1', 'p_x4y2z1']], 
      dtype='|S8')
于 2013-10-15T13:45:34.400 回答