matplotlib의 등고선 그리기 속도를 높이십시오

Question

파일의 데이터를 텍스트 파일의 각 행에 대한 등고선 그래프로 표시하는 Python 프로그램이 있습니다. 현재 인터페이스에 3 개의 별도 윤곽 플롯이 있습니다. 파일에서 데이터를 읽거나 스크립트를 실행하기 전에 메모리에로드하면 컨투어 플롯에서 ~ 6fps 밖에 얻을 수 없습니다.

나는 하나의 등고선을 사용하고 나머지는 일반 플롯을 사용했지만 속도는 단지 7fps로 증가했습니다. 나는 너무 계산적으로 몇 줄을 그어야한다고 생각하지 않는다. 실질적으로 더 빨리 만들 수있는 방법이 있습니까? 적어도 30fps를 얻는 것이 이상적입니다.

내가 윤곽을 그리는 방법은 내 데이터의 각 행에 대해 내가 이전 제거하는 것이 있습니다 :

for coll in my_contour[0].collections: 
    coll.remove() 

및 코드의 시작 부분에

하나의 새로운

my_contour[0] = ax[0].contour(x, y, my_func, [0]) 

을 추가, 플롯을 추가 할 때 plt.ion()을 업데이트해야합니다.

도움을 주시면 감사하겠습니다.

감사

Answer 1

다음은 애니메이션에 contour 플롯을 사용하는 방법에 대한 예입니다. 블리 팅을 쉽게 켜거나 끌 수있는 matplotlib.animation.FuncAnimation을 사용합니다. blit = True로 설정하면 ~ 55fps를 블리핑하지 않고 내 컴퓨터에서 ~ 64fps로 실행됩니다. 물론 interval은 빠른 애니메이션을 허용해야합니다. interval=10 (밀리 초)으로 설정하면 최대 100fps가 허용되지만 그리기 시간은 이보다 느린 속도로 제한됩니다. 이미지를 저장하는 과정이 좀 더 걸리기 때문에, 더 느린 프레임 속도를 상기 애니메이션 GIF에 도시되어

는

import matplotlib.pyplot as plt 
import matplotlib.animation 
import numpy as np 
import time 

x= np.linspace(0,3*np.pi) 
X,Y = np.meshgrid(x,x) 
f = lambda x,y, alpha, beta :(np.sin(X+alpha)+np.sin(Y*(1+np.sin(beta)*.4)+alpha))**2 
alpha=np.linspace(0, 2*np.pi, num=34) 
levels= 10 
cmap=plt.cm.magma 


fig, ax=plt.subplots() 
props = dict(boxstyle='round', facecolor='wheat') 
timelabel = ax.text(0.9,0.9, "", transform=ax.transAxes, ha="right", bbox=props) 
t = np.ones(10)*time.time() 
p = [ax.contour(X,Y,f(X,Y,0,0), levels, cmap=cmap) ] 

def update(i): 
    for tp in p[0].collections: 
     tp.remove() 
    p[0] = ax.contour(X,Y,f(X,Y,alpha[i],alpha[i]), levels, cmap= cmap) 
    t[1:] = t[0:-1] 
    t[0] = time.time() 
    timelabel.set_text("{:.3f} fps".format(-1./np.diff(t).mean())) 
    return p[0].collections+[timelabel] 

ani = matplotlib.animation.FuncAnimation(fig, update, frames=len(alpha), 
             interval=10, blit=True, repeat=True) 
plt.show() 

참고.