matplotlib plt.figure ist ein Numpy-Array

Idee: Verwenden Sie plt.savefig, um Dateien im BytesIO-Cache zu speichern, und verwenden Sie dann PIL, um die Dateien im Cache zu lesen und in Numpy zu konvertieren

from PIL import Image
from io import BytesIO

img = np.ones((256,256,3), dtype=np.uint8)
plt.figure()
plt.imshow(img, cmap='gray')
buffer = BytesIO()
plt.savefig(buffer, format='png')
new_img = np.asarray(Image.open(buffer)) # new_img就是figure的数组
plt.close()
buffer.close()