Matplotlib: lưu cốt truyện thành mảng có nhiều mảng

Question

Trong Python và Matplotlib, thật dễ dàng hiển thị ô như cửa sổ bật lên hoặc lưu cốt truyện dưới dạng tệp PNG. Làm thế nào tôi có thể thay thế lưu âm mưu vào một mảng numpy ở định dạng RGB?

Answer 1

Đây là mẹo hữu ích cho các bài kiểm tra đơn vị và tương tự, khi bạn cần thực hiện so sánh pixel-điểm ảnh với một ô đã lưu.

Một cách là sử dụng fig.canvas.tostring_rgb và sau đó là numpy.fromstring với kiểu chữ approriate. Có nhiều cách khác nữa, nhưng đây là cách tôi có xu hướng sử dụng.

Ví dụ:

import matplotlib.pyplot as plt 
import numpy as np 

# Make a random plot... 
fig = plt.figure() 
fig.add_subplot(111) 

# If we haven't already shown or saved the plot, then we need to 
# draw the figure first... 
fig.canvas.draw() 

# Now we can save it to a numpy array. 
data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='') 
data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))