ParaView：使用Programmable Filter创建新向量报错及正确性问询

问题描述 #

加载VTI文件后，需基于名为MyVelocity的3D向量数组，按以下规则生成新向量：

unew = -u
vnew = -v
wnew =  w

使用Programmable Filter编写代码后触发类型转换错误，原代码及报错信息如下：

原代码 #

import numpy

u=inputs[0].PointData["MyVelocity"][:,0]
v=inputs[0].PointData["MyVelocity"][:,1]
w=inputs[0].PointData["MyVelocity"][:,2]
output.PointData.append([-u,-v,w], "vector")

报错信息 #

Traceback (most recent call last):
  File "<string>", line 22, in <module>
  File "<string>", line 7, in RequestData
  File "C:\Program Files\ParaView 5.13.0\bin\Lib\site-packages\vtkmodules\numpy_interface\dataset_adapter.py", line 763, in append
    arr = numpyTovtkDataArray(copy, name)
  File "C:\Program Files\ParaView 5.13.0\bin\Lib\site-packages\vtkmodules\numpy_interface\dataset_adapter.py", line 146, in numpyTovtkDataArray
    vtkarray = numpy_support.numpy_to_vtk(array, array_type=array_type)
  File "C:\Program Files\ParaView 5.13.0\bin\Lib\site-packages\vtkmodules\util\numpy_support.py", line 146, in numpy_to_vtk
    vtk_typecode = get_vtk_array_type(z.dtype)
  File "C:\Program Files\ParaView 5.13.0\bin\Lib\site-packages\vtkmodules\util\numpy_support.py", line 69, in get_vtk_array_type
    raise TypeError(
TypeError: Could not find a suitable VTK type for object

错误原因 #

代码中[-u,-v,w]生成的是包含三个一维numpy数组的列表，而ParaView的output.PointData.append()要求输入形状为(N,3)的二维numpy数组（N为点总数），列表格式无法被VTK正确解析，因此抛出类型转换错误。

正确实现代码 #

提供两种可行的实现方式：

方式一：直接修改原向量副本 #

import numpy

# 获取原始3D向量数组
original_vel = inputs[0].PointData["MyVelocity"]
# 复制数组避免修改原始数据
new_vel = original_vel.copy()
# 按规则修改分量：u和v取反，w保持不变
new_vel[:, 0] = -new_vel[:, 0]
new_vel[:, 1] = -new_vel[:, 1]
# 将新向量添加到输出点数据中
output.PointData.append(new_vel, "NewVelocity")

方式二：合并处理后的分量数组 #

import numpy

# 提取原始向量的三个分量
u = inputs[0].PointData["MyVelocity"][:, 0]
v = inputs[0].PointData["MyVelocity"][:, 1]
w = inputs[0].PointData["MyVelocity"][:, 2]
# 将处理后的分量合并为(N,3)的二维数组
new_vel = numpy.column_stack((-u, -v, w))
# 添加新向量到输出
output.PointData.append(new_vel, "NewVelocity")

验证方法 #

代码运行后，可通过以下方式验证结果：

• 在ParaView的**信息(Information)**面板中，查看输出数据集的点数据，确认NewVelocity为3D向量类型；
• 使用Glyph过滤器可视化向量，检查方向是否符合unew=-u, vnew=-v, wnew=w的规则。

内容的提问来源于stack exchange，提问作者shashashamti2008