numpy 配列を .vti形式で書き出す

需要があるかわかりませんが,自分の備忘録として書きます.今回はtvtkというライブラリで実現しています.前回の記事でvtkArray形式に変換できているので,普通にvtkで保存してもいいと思います.

   

以下,インポート

from tvtk.api import tvtk, write_data

以下,メソッド

def write_vtk_file_from_ndarray(dim3_array, output_path, spacing=(1, 1, 1), data_name="Hoge"):
    grid = tvtk.ImageData(spacing=spacing, origin=(0, 0, 0), dimensions=dim3_array.shape)
    grid.point_data.scalars = dim3_array.ravel(order='F')
    grid.point_data.scalars.name = data_name
    write_data(grid, output_path)
   
カテゴリー: プログラミング タグ: , , パーマリンク