matplotlib rất chậm. Nó có bình thường không?

Question

Tôi đang tạo một vài ô pdf với matplotlib bao gồm 400 ô con. Mỗi người chỉ có 5 điểm dữ liệu. Phải mất 420 s trên một máy tính tốt để tiết kiệm 5 hình ảnh pdf. Có cách nào để tối ưu hóa mã hoặc nó chỉ là bình thường đối với matplotlib?

Phần mã cho âm mưu:

plot_cnt = 1 
for k in np.arange(K_min, K_max + 1): 
    for l in np.arange(L_min, L_max + 1): 
     ax = plt.subplot(grid[0], grid[1], plot_cnt) 
     plot_cnt += 1 
     plt.setp(ax, 'frame_on', False) 
     ax.set_ylim([-0.1, 1.1]) 
     ax.set_xlabel('K={},L={}'.format(k, l), size=3) 
     ax.set_xlim([-0.1, 4.1]) 
     ax.set_xticks([]) 
     ax.set_yticks([]) 
     ax.grid('off') 
     ax.plot(np.arange(5), (data['S1']['Azimuth'][:, k - 1, l + offset_l] + \ 
       data['S1']['Delta Speed'][:, k - 1, l + offset_l] + \ 
       data['S1']['Speed'][:, k - 1, l + offset_l])/3, 
       'r-o', ms=1, mew=0, mfc='r') 
     ax.plot(np.arange(5), data['S2'][case][:, k - 1, l + offset_l], 
       'b-o', ms=1, mew=0, mfc='b') 
plt.savefig(os.path.join(os.getcwd(), 'plot-average.pdf')) 
plt.clf() 
print 'Final plot created.' 

Ảnh cuối cùng:

Answer 1

Building tắt của những gì @rowman nói, bạn có thể làm điều này tất cả trong một trục (như bạn tắt tất cả các ve v.v.) Một cái gì đó như:

K_max = 20 
K_min = 0 
L_max = 20 
L_min = 0 
ax = plt.subplot(111) 
x_offset = 7 # tune these 
y_offset = 7 # tune these 
plt.setp(ax, 'frame_on', False) 
ax.set_ylim([0, (K_max-K_min +1)*y_offset ]) 
ax.set_xlim([0, (L_max - L_min+1)*x_offset]) 
ax.set_xticks([]) 
ax.set_yticks([]) 
ax.grid('off') 



for k in np.arange(K_min, K_max + 1): 
    for l in np.arange(L_min, L_max + 1): 
     ax.plot(np.arange(5) + l*x_offset, 5+rand(5) + k*y_offset, 
       'r-o', ms=1, mew=0, mfc='r') 
     ax.plot(np.arange(5) + l*x_offset, 3+rand(5) + k*y_offset, 
       'b-o', ms=1, mew=0, mfc='b') 
     ax.annotate('K={},L={}'.format(k, l), (2.5+ (k)*x_offset,l*y_offset), size=3,ha='center') 
plt.savefig(os.path.join(os.getcwd(), 'plot-average.pdf')) 

print 'Final plot created.' 

Chạy trong khoảng một hoặc hai giây. Tôi nghĩ rằng tất cả thời gian được dành cho việc thiết lập đối tượng axes khá phức tạp trong nội bộ.