[파이썬] STL 3차원 그래프 그리기 (pyvista)

STL 파일을 불러와서 3차원에 시각화하려고 하는데 matplotlib 로 그리니까 너무 느리다. 심각하게 느려서 View 변경하는게 거의 불가능한 정도였다. 

다른 방법을 찾던 중 pyvista 라는 패키지를 알게 되었다. 매쉬 데이터의 3차원 시각화가 목적인 패키지다. 써봤는데 엄청빠르다. 

아래와 같은 형식으로 사용한다. pv.read 를 이용하여 stl을 불러오고 Plotter 함수를 이용하여 그려준다. 

import os
import pyvista as pv

path="C:/Users/python_stl"

os.chdir(path)
cat = pv.read('cat.stl')

# Create a plot
p = pv.Plotter() # 캔버스 정의 
p.add_mesh(cat, opacity=1, color='red')
p.show()